def test_iterable_flat(self):
node1 = _create_node('1')
node2 = _create_node('2')
node3 = _create_node('3')
expected = _create_nodes('1', '2', '3')
self.assertEqual(expected, phase_collections.flatten([node1, node2, node3]))
def test_iterable_of_iterable(self):
nodes = [[_create_node('1')],
[[_create_node('2'), _create_node('3')], [_create_node('4')],
_create_node('5')],
_create_node('6'), phase]
expected = _create_nodes('1', '2', '3', '4', '5', '6', 'phase')
self.assertEqual(expected, phase_collections.flatten(nodes))
def test_single_node(self):
node = _create_node('a')
expected = _create_nodes('a')
self.assertEqual(expected, phase_collections.flatten(node))
def test_flatten_single_list(self):
seq = htf.PhaseSequence(_create_nodes('1', '2'))
expected = [htf.PhaseSequence(_create_nodes('1', '2'))]
self.assertEqual(expected, phase_collections.flatten([seq]))
def test_invalid_entry(self):
nodes = 42
with self.assertRaises(ValueError):
phase_collections.flatten(nodes)
def test_single_phase(self):
expected = _create_nodes('phase')
self.assertEqual(expected, phase_collections.flatten(phase))
