Add a test with custom classes

Author: lesteve

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"""