Simpler test setup.
This commit is contained in:
@@ -794,24 +794,18 @@ class TestSequence(unittest.TestCase):
|
|||||||
|
|
||||||
def test_flatten(self):
|
def test_flatten(self):
|
||||||
node = DummySchemaNode(None, name='node')
|
node = DummySchemaNode(None, name='node')
|
||||||
int1 = DummyType()
|
|
||||||
int2 = DummyType()
|
|
||||||
node.children = [
|
node.children = [
|
||||||
DummySchemaNode(int1, name='a'),
|
DummySchemaNode(DummyType(), name='foo'),
|
||||||
DummySchemaNode(int2, name='b'),
|
]
|
||||||
]
|
|
||||||
typ = self._makeOne()
|
typ = self._makeOne()
|
||||||
result = typ.flatten(node, [1, 2])
|
result = typ.flatten(node, [1, 2])
|
||||||
self.assertEqual(result, {'node.0': 1, 'node.1': 2})
|
self.assertEqual(result, {'node.0': 1, 'node.1': 2})
|
||||||
|
|
||||||
def test_flatten_listitem(self):
|
def test_flatten_listitem(self):
|
||||||
node = DummySchemaNode(None, name='node')
|
node = DummySchemaNode(None, name='node')
|
||||||
int1 = DummyType()
|
|
||||||
int2 = DummyType()
|
|
||||||
node.children = [
|
node.children = [
|
||||||
DummySchemaNode(int1, name='a'),
|
DummySchemaNode(DummyType(), name='foo'),
|
||||||
DummySchemaNode(int2, name='b'),
|
]
|
||||||
]
|
|
||||||
typ = self._makeOne()
|
typ = self._makeOne()
|
||||||
result = typ.flatten(node, [1, 2], listitem=True)
|
result = typ.flatten(node, [1, 2], listitem=True)
|
||||||
self.assertEqual(result, {'0': 1, '1': 2})
|
self.assertEqual(result, {'0': 1, '1': 2})
|
||||||
|
Reference in New Issue
Block a user