Exemplo n.º 1
0
 def test_no_nesting(self):
     self.assertEqual(['a', 'bcd', 123, 45.6],
                      list(tools.flatten('a', 'bcd', 123, 45.6)))
     self.assertEqual(['a', 'bcd', 123, 45.6],
                      list(tools.flatten(['a', 'bcd', 123, 45.6])))
     self.assertEqual(['a', 'bcd', 123, 45.6],
                      list(tools.flatten(['a', 'bcd'], [123, 45.6])))
Exemplo n.º 2
0
 def test_nesting(self):
     self.assertEqual(['a', 'bcd', 123, 45.6],
                      list(tools.flatten('a', ['bcd'], [123], 45.6)))
     self.assertEqual(['a', 'bcd', 123, 45.6],
                      list(tools.flatten([['a', ('bcd', [123])], 45.6])))
     self.assertEqual(['a', 'bcd', 123, 45.6],
                      list(tools.flatten([('a', 'bcd')],
                                         [[[[123]]], 45.6])))
Exemplo n.º 3
0
 def __init__(self, name: str, steps: Iterable[Callable]):
     self.name = name
     self._steps = tuple(tools.flatten(steps))
Exemplo n.º 4
0
 def test_empty(self):
     self.assertEqual([], list(tools.flatten()))
     self.assertEqual([], list(tools.flatten([])))
     self.assertEqual([], list(tools.flatten([], ())))
     self.assertEqual([], list(tools.flatten([[], (), [[]]], ((), []))))