def test_flatten(self): out1 = sm.flatten(self.input) out2 = sm.flatten(self.input, unique=False) self.assertEquals(out1, [1, 2, 3, 4, 5, 6]) self.assertEquals(out2, [1, 2, 3, 3, 4, 5, 6])
def test_flatten(self): out1 = sm.flatten(self.input) out2 = sm.flatten(self.input, unique=False) self.assertEquals(out1, [1, 2, 3, 4, 5, 6]) self.assertEquals(out2, [1, 2, 3, 3, 4, 5, 6])