def test_wrap_objects(self): class A(object): a = 1 l = [A(), A(), A()] self.assertIsInstance(_wrap(A()), A) self.assert_type(self.seq(l))
def test_wrap(self): self.assert_type(_wrap([1, 2])) self.assert_type(_wrap((1, 2))) self.assert_not_type(_wrap(1)) self.assert_not_type(_wrap(1.0)) self.assert_not_type(_wrap("test")) self.assert_not_type(_wrap(True)) self.assert_not_type(_wrap(Data(1, 2)))