Import only necessary single TestCase, not the whole module

diff --git a/pyac/tests/test_blob.py b/pyac/tests/test_blob.py
index ffef43ac3bf7956d62a9f09d4e6650524756f7c566bd720840acd933b964132d..152d3d784180da5f1f4782003a1d514bc5bbda82a234aca2afc20c09e16b93d5 100644 (file)
--- a/pyac/tests/test_blob.py
+++ b/pyac/tests/test_blob.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from hypothesis import given
 from hypothesis.strategies import binary
@@ -10,7 +10,7 @@ from pyac import loads
 from pyac import NotEnoughData
 
 
-class TestBlob(unittest.TestCase):
+class TestBlob(TestCase):
     def test_blob_encode(self) -> None:
         blob = Blob(4, b"testdata")
         encoded = dumps(blob)

diff --git a/pyac/tests/test_bool.py b/pyac/tests/test_bool.py
index 586deb714e898be94a03284361bd1fce1086bdbdc0544446d4b008d77e738557..4e0766ee7eae0ca351a6dc1c1748f092371cb0a2f97cda2bfc4d4d13f9b06675 100644 (file)
--- a/pyac/tests/test_bool.py
+++ b/pyac/tests/test_bool.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from hypothesis import given
 from hypothesis.strategies import binary
@@ -7,7 +7,7 @@ from pyac import dumps
 from pyac import loads
 
 
-class TestBool(unittest.TestCase):
+class TestBool(TestCase):
     def test_bool_encode_true(
         self,
     ):

diff --git a/pyac/tests/test_errors.py b/pyac/tests/test_errors.py
index b816910b199206c37776959c3175851de8e0ae11cce7fd6f37381ae15da0401a..7fa8801080f6b31402a8f36fa8163a9e2db2445b6f9305fcc4911480d481e297 100644 (file)
--- a/pyac/tests/test_errors.py
+++ b/pyac/tests/test_errors.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from hypothesis import given
 from hypothesis.strategies import integers
@@ -10,7 +10,7 @@ from pyac import loads
 from pyac import NotEnoughData
 
 
-class TestError(unittest.TestCase):
+class TestError(TestCase):
     @given(integers(min_value=1, max_value=100))
     def test_not_enough_data_str(self, integer: int) -> None:
         self.assertEqual(str(NotEnoughData(integer)), "{} bytes expected".format(integer))

diff --git a/pyac/tests/test_float.py b/pyac/tests/test_float.py
index cccf4606cbfe45d99ad758312ed6b8fd15c1525cdc3944c3bd6e8ee6844f169b..1dfc269416ad82ede7e3adf78e13d9df67a8e73c2852fc3a6bd4ca62c88aed37 100644 (file)
--- a/pyac/tests/test_float.py
+++ b/pyac/tests/test_float.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from typing import List
 
@@ -8,7 +8,7 @@ from pyac import NotEnoughData
 from pyac import Raw
 
 
-class TestFloat(unittest.TestCase):
+class TestFloat(TestCase):
     def test_throws_when_dumps_float(self) -> None:
         with self.assertRaises(NotImplementedError) as cm:
             dumps(1.5)

diff --git a/pyac/tests/test_int.py b/pyac/tests/test_int.py
index 768a6c8486c712a108e7d4da49b1e43ac270806ef1c28319036b795efeafb449..ca5154c5fac1198cebf3b9b8076dfe94c7af28021d1d1ff969183136d6ce542c 100644 (file)
--- a/pyac/tests/test_int.py
+++ b/pyac/tests/test_int.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from pyac import dumps
 from pyac import DecodeError
@@ -6,7 +6,7 @@ from pyac import loads
 from pyac import NotEnoughData
 
 
-class TestInt(unittest.TestCase):
+class TestInt(TestCase):
     def test_int_positive(self) -> None:
         ints: list[int] = [1, 123, 1 << 64, 1 << 130]
         expected: list[bytes] = [

diff --git a/pyac/tests/test_list.py b/pyac/tests/test_list.py
index 6fffd0485a309b0a1d1a4f76912bea74399383b2c98fe325a0e8d06f3d8c742d..55453e2520f8afd4133c66f6fffa56a5914cd86bf1c83237682f8305ef0b374b 100644 (file)
--- a/pyac/tests/test_list.py
+++ b/pyac/tests/test_list.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from hypothesis import given
 from hypothesis.strategies import binary
@@ -43,7 +43,7 @@ any_st = one_of(
 )
 
 
-class TestList(unittest.TestCase):
+class TestList(TestCase):
     def test_list_encode_empty(self) -> None:
         encoded = dumps([])
         self.assertEqual(encoded, b"\x08\x00")

diff --git a/pyac/tests/test_map.py b/pyac/tests/test_map.py
index afaac018c448144e92505627cbf5f42954b966fb032307423c0248bf762d687a..b8c4d546d94b2213d7e2d19cc7904a65da62b54fcfac03ad19769a74568beb58 100644 (file)
--- a/pyac/tests/test_map.py
+++ b/pyac/tests/test_map.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from hypothesis import given
 from hypothesis.strategies import binary
@@ -46,7 +46,7 @@ mapkey_st = text(
 )
 
 
-class TestMap(unittest.TestCase):
+class TestMap(TestCase):
     @given(dictionaries(keys=mapkey_st, values=any_st, max_size=4))
     def test_map_encode(self, test_map):
         encoded = dumps(test_map)

diff --git a/pyac/tests/test_str.py b/pyac/tests/test_str.py
index fd594be137c12d9f72b9ca0959ebafb8900b5f48c8020624879a1aaae4409054..5b0824fd32ee325c3288ad9c64bb57c8e58c253eb7d7b206fab01f9f712994a2 100644 (file)
--- a/pyac/tests/test_str.py
+++ b/pyac/tests/test_str.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from hypothesis import given
 from hypothesis.strategies import integers
@@ -25,7 +25,7 @@ invalid_utf8_3byte_s = integers(min_value=0, max_value=(1 << 10) - 1).map(
 )
 
 
-class TestString(unittest.TestCase):
+class TestString(TestCase):
     @given(text(max_size=60))
     def test_encode_utf8(self, test_str: str) -> None:
         if(len(test_str.encode("utf-8")) > 60):

diff --git a/pyac/tests/test_symmetric.py b/pyac/tests/test_symmetric.py
index 6cf674f49514af40a92f9d9920f9be2f0ba709674c6809cec803813908085d57..62db02dd8f35b098e641cd29fc9eb32f4bdcdf2b1603267252fdd9ab76fb120d 100644 (file)
--- a/pyac/tests/test_symmetric.py
+++ b/pyac/tests/test_symmetric.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from typing import Any
 from hypothesis import given
@@ -57,7 +57,7 @@ everything_st = deferred(
 )
 
 
-class TestSymmetric(unittest.TestCase):
+class TestSymmetric(TestCase):
     @given(everything_st, binary(max_size=20))
     def test_symmetric(self, obj: Any, junk: bytes) -> None:
         encoded: bytes = dumps(obj) + junk

diff --git a/pyac/tests/test_tai.py b/pyac/tests/test_tai.py
index 9fd24f478b53e3397154f58235d7a53722f8b97d9e208cb668471199c32d1967..6af631de744b5b4c96f6a50f10cff71ccb067b4b62df9b1b4383a0f2d73fbb3f 100644 (file)
--- a/pyac/tests/test_tai.py
+++ b/pyac/tests/test_tai.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from datetime import datetime
 from datetime import timedelta
@@ -49,7 +49,7 @@ DJB_Leapsecs = (
 )
 
 
-class TestTAI64(unittest.TestCase):
+class TestTAI64(TestCase):
     @given(binary(max_size=20))
     def test_encode_decode_tai64(self, junk: bytes) -> None:
         dt = datetime(2023, 10, 1, 12, 0, 0)

diff --git a/pyac/tests/test_uuid.py b/pyac/tests/test_uuid.py
index 806f6640d48503356b6ce7374f9ae1e78563954db01bbf043ce8b4cb0b375ba8..4fe2eb6c75fd0f75330e504b0abd201ecda2910331aaee1ca6409c0de88c4e30 100644 (file)
--- a/pyac/tests/test_uuid.py
+++ b/pyac/tests/test_uuid.py
@@ -1,4 +1,4 @@
-import unittest
+from unittest import TestCase
 
 from uuid import UUID
 
@@ -7,7 +7,7 @@ from pyac import loads
 from pyac import NotEnoughData
 
 
-class TestUUID(unittest.TestCase):
+class TestUUID(TestCase):
     def test_uuid_encode(self) -> None:
         uuid_str: str = "12345678-1234-5678-1234-567812345678"
         uuid_obj: UUID = UUID(uuid_str)