def test_flatten(self): self.assertListEqual( flatten([1, 2, [3, 4], (5, 6)]), [1, 2, 3, 4, 5, 6]) self.assertListEqual( flatten([[[1, 2, 3], (42, None)], [4, 5], [6], 7, (8, 9, 10)]), [1, 2, 3, 42, None, 4, 5, 6, 7, 8, 9, 10])
def select(self, xpath): return self.__class__(flatten([x.select(xpath) for x in self]))
def test_flatten(self): self.assertListEqual(flatten([1, 2, [3, 4], (5, 6)]), [1, 2, 3, 4, 5, 6]) self.assertListEqual( flatten([[[1, 2, 3], (42, None)], [4, 5], [6], 7, (8, 9, 10)]), [1, 2, 3, 42, None, 4, 5, 6, 7, 8, 9, 10])