Commit e78592d4 authored by Serhiy Storchaka's avatar Serhiy Storchaka

Make bytes and bytearray subclass tests compatible with base types tests.

parent ea36c941
...@@ -1425,11 +1425,11 @@ class BytesAsStringTest(FixedStringTest, unittest.TestCase): ...@@ -1425,11 +1425,11 @@ class BytesAsStringTest(FixedStringTest, unittest.TestCase):
class SubclassTest: class SubclassTest:
def test_basic(self): def test_basic(self):
self.assertTrue(issubclass(self.subclass2test, self.type2test)) self.assertTrue(issubclass(self.type2test, self.basetype))
self.assertIsInstance(self.subclass2test(), self.type2test) self.assertIsInstance(self.type2test(), self.basetype)
a, b = b"abcd", b"efgh" a, b = b"abcd", b"efgh"
_a, _b = self.subclass2test(a), self.subclass2test(b) _a, _b = self.type2test(a), self.type2test(b)
# test comparison operators with subclass instances # test comparison operators with subclass instances
self.assertTrue(_a == _a) self.assertTrue(_a == _a)
...@@ -1452,19 +1452,19 @@ class SubclassTest: ...@@ -1452,19 +1452,19 @@ class SubclassTest:
# Make sure join returns a NEW object for single item sequences # Make sure join returns a NEW object for single item sequences
# involving a subclass. # involving a subclass.
# Make sure that it is of the appropriate type. # Make sure that it is of the appropriate type.
s1 = self.subclass2test(b"abcd") s1 = self.type2test(b"abcd")
s2 = self.type2test().join([s1]) s2 = self.basetype().join([s1])
self.assertTrue(s1 is not s2) self.assertTrue(s1 is not s2)
self.assertTrue(type(s2) is self.type2test, type(s2)) self.assertTrue(type(s2) is self.basetype, type(s2))
# Test reverse, calling join on subclass # Test reverse, calling join on subclass
s3 = s1.join([b"abcd"]) s3 = s1.join([b"abcd"])
self.assertTrue(type(s3) is self.type2test) self.assertTrue(type(s3) is self.basetype)
def test_pickle(self): def test_pickle(self):
a = self.subclass2test(b"abcd") a = self.type2test(b"abcd")
a.x = 10 a.x = 10
a.y = self.subclass2test(b"efgh") a.y = self.type2test(b"efgh")
for proto in range(pickle.HIGHEST_PROTOCOL + 1): for proto in range(pickle.HIGHEST_PROTOCOL + 1):
b = pickle.loads(pickle.dumps(a, proto)) b = pickle.loads(pickle.dumps(a, proto))
self.assertNotEqual(id(a), id(b)) self.assertNotEqual(id(a), id(b))
...@@ -1475,9 +1475,9 @@ class SubclassTest: ...@@ -1475,9 +1475,9 @@ class SubclassTest:
self.assertEqual(type(a.y), type(b.y)) self.assertEqual(type(a.y), type(b.y))
def test_copy(self): def test_copy(self):
a = self.subclass2test(b"abcd") a = self.type2test(b"abcd")
a.x = 10 a.x = 10
a.y = self.subclass2test(b"efgh") a.y = self.type2test(b"efgh")
for copy_method in (copy.copy, copy.deepcopy): for copy_method in (copy.copy, copy.deepcopy):
b = copy_method(a) b = copy_method(a)
self.assertNotEqual(id(a), id(b)) self.assertNotEqual(id(a), id(b))
...@@ -1487,6 +1487,8 @@ class SubclassTest: ...@@ -1487,6 +1487,8 @@ class SubclassTest:
self.assertEqual(type(a), type(b)) self.assertEqual(type(a), type(b))
self.assertEqual(type(a.y), type(b.y)) self.assertEqual(type(a.y), type(b.y))
test_fromhex = BaseBytesTest.test_fromhex
class ByteArraySubclass(bytearray): class ByteArraySubclass(bytearray):
pass pass
...@@ -1498,8 +1500,8 @@ class OtherBytesSubclass(bytes): ...@@ -1498,8 +1500,8 @@ class OtherBytesSubclass(bytes):
pass pass
class ByteArraySubclassTest(SubclassTest, unittest.TestCase): class ByteArraySubclassTest(SubclassTest, unittest.TestCase):
type2test = bytearray basetype = bytearray
subclass2test = ByteArraySubclass type2test = ByteArraySubclass
def test_init_override(self): def test_init_override(self):
class subclass(bytearray): class subclass(bytearray):
...@@ -1513,8 +1515,8 @@ class ByteArraySubclassTest(SubclassTest, unittest.TestCase): ...@@ -1513,8 +1515,8 @@ class ByteArraySubclassTest(SubclassTest, unittest.TestCase):
class BytesSubclassTest(SubclassTest, unittest.TestCase): class BytesSubclassTest(SubclassTest, unittest.TestCase):
type2test = bytes basetype = bytes
subclass2test = BytesSubclass type2test = BytesSubclass
if __name__ == "__main__": if __name__ == "__main__":
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment