diff --git a/tests/protocols/test_generic.py b/tests/protocols/test_generic.py
index 088c8ee..4ec61f0 100644
--- a/tests/protocols/test_generic.py
+++ b/tests/protocols/test_generic.py
@@ -12,7 +12,7 @@
 logging.basicConfig(level=logging.DEBUG)
 
 
-class TestSwitch1(unittest.TestCase):
+class TestGeneric(unittest.TestCase):
     def test_decode_1(self) -> None:
         decoded = generic.decode(
             "020102010201020101020102010201020102020101020201020102010102020101020201010202010201020102010201020102010201020102010201020102010201020102010201020102010201020102010201020102010201020102010201010203"
diff --git a/tests/protocols/test_switch1.py b/tests/protocols/test_switch1.py
index 04886cd..b7efc15 100644
--- a/tests/protocols/test_switch1.py
+++ b/tests/protocols/test_switch1.py
@@ -17,7 +17,9 @@ def test_decode_1(self) -> None:
         decoded = switch1.decode(
             "020100010001000100010001000100010001000100010001000100010001000100010001000100010001000100010001000100010000010100000100010001000103"
         )
-        assert decoded == {"id": 67108863, "all": False, "state": True, "unit": 0}
+        self.assertDictEqual(
+            {"id": 67108863, "all": False, "state": True, "unit": 0}, decoded
+        )
 
     def test_decode_2(self) -> None:
         decoded = switch1.decode(
diff --git a/tests/protocols/test_switch3.py b/tests/protocols/test_switch3.py
index 9ae9224..77e87e6 100644
--- a/tests/protocols/test_switch3.py
+++ b/tests/protocols/test_switch3.py
@@ -23,7 +23,7 @@ def test_decode_1(self) -> None:
     #
     # def test_decode_3(self) -> None:
     #     decoded = switch3.decode('01010110010101100110011001100110010101100110011002')
-    #     assert decoded == {'unit': 20, 'id': 2, 'state': True}
+    #     self.assertDictEqual({'unit': 20, 'id': 2, 'state': True}, decoded)
 
     def test_encode_1(self) -> None:
         encoded = switch3.encode(id=25, unit=16, state=False)