Fix type in assertRaisesMessage signature

pytest-dev · Apr 26, 2022 · 5c6c937 · 5c6c937
1 parent f27c6aa
commit 5c6c937
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytest_django/asserts.py b/pytest_django/asserts.py
@@ -2,7 +2,7 @@
 Dynamically load all Django assertion cases and expose them for importing.
 """
 from functools import wraps
-from typing import Any, Callable, Optional, Sequence, Set, Union
+from typing import Any, Callable, Optional, Sequence, Set, Type, Union
 
 from django.test import (
     LiveServerTestCase, SimpleTestCase, TestCase, TransactionTestCase,
@@ -113,7 +113,7 @@ def assertTemplateNotUsed(
         ...
 
     def assertRaisesMessage(
-        expected_exception: BaseException,
+        expected_exception: Type[Exception],
         expected_message: str,
         *args,
         **kwargs