Add a test with custom classes

pytest-dev · Dec 29, 2016 · babbff4 · babbff4
1 parent 6173120
commit babbff4
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/testing/test_recwarn.py b/testing/test_recwarn.py
@@ -247,6 +247,17 @@ def test_record_by_subclass(self):
         assert str(record[0].message) == "user"
         assert str(record[1].message) == "runtime"
 
+        class MyUserWarning1(UserWarning): pass
+        class MyUserWarning2(UserWarning): pass
+
+        with pytest.warns(UserWarning) as record:
+            warnings.warn("user 1", MyUserWarning1)
+            warnings.warn("user 2", MyUserWarning2)
+
+        assert len(record) == 2
+        assert str(record[0].message) == "user 1"
+        assert str(record[1].message) == "user 2"
+
 
     def test_double_test(self, testdir):
         """If a test is run again, the warning should still be raised"""