Spaces:
Running
Running
File size: 1,525 Bytes
122d3ff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
from .common import TestCase
class TestCompletions(TestCase):
def test_group_completions(self):
# Test completions on top-level file.
g = self.f.create_group('g')
self.f.create_group('h')
self.f.create_dataset('data', [1, 2, 3])
self.assertEqual(
self.f._ipython_key_completions_(),
['data', 'g', 'h'],
)
self.f.create_group('data2', [1, 2, 3])
self.assertEqual(
self.f._ipython_key_completions_(),
['data', 'data2', 'g', 'h'],
)
# Test on subgroup.
g.create_dataset('g_data1', [1, 2, 3])
g.create_dataset('g_data2', [4, 5, 6])
self.assertEqual(
g._ipython_key_completions_(),
['g_data1', 'g_data2'],
)
g.create_dataset('g_data3', [7, 8, 9])
self.assertEqual(
g._ipython_key_completions_(),
['g_data1', 'g_data2', 'g_data3'],
)
def test_attrs_completions(self):
attrs = self.f.attrs
# Write out of alphabetical order to test that completions come back in
# alphabetical order, as opposed to, say, insertion order.
attrs['b'] = 1
attrs['a'] = 2
self.assertEqual(
attrs._ipython_key_completions_(),
['a', 'b']
)
attrs['c'] = 3
self.assertEqual(
attrs._ipython_key_completions_(),
['a', 'b', 'c']
)
|