Fix test failures

jelmer · Nov 20, 2024 · c6420e7 · c6420e7
1 parent 5bd188b
commit c6420e7
Showing 1 changed file with 6 additions and 15 deletions.
diff --git a/dulwich/tests/test_object_store.py b/dulwich/tests/test_object_store.py
@@ -20,7 +20,7 @@
 
 """Tests for the object store interface."""
 
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Callable, Any
 from unittest import skipUnless
 
 from dulwich.index import commit_tree
@@ -53,20 +53,11 @@
 class ObjectStoreTests:
     store: "BaseObjectStore"
 
-    def assertEqual(self, a, b) -> None:
-        raise NotImplementedError
-
-    def assertRaises(self, exc, func) -> None:
-        raise NotImplementedError
-
-    def assertNotIn(self, a, b) -> None:
-        raise NotImplementedError
-
-    def assertNotEqual(self, a, b) -> None:
-        raise NotImplementedError
-
-    def assertIn(self, a, b) -> None:
-        raise NotImplementedError
+    assertEqual: Callable[[object, object], None]
+    assertRaises: Callable[[type[Exception], Callable[[], Any]], None]
+    assertNotIn: Callable[[object, object], None]
+    assertNotEqual: Callable[[object, object], None]
+    assertIn: Callable[[object, object], None]
 
     def test_determine_wants_all(self) -> None:
         self.assertEqual(