test: improve tests for @queries_records #296

- Previously, asserting the status code was unreachable

test/test_queries_records.py

@@ -30,7 +30,10 @@ def test_valid_error_raised(
         def raise_exception():
             raise raised_exception()
 
-        with pytest.raises(expected_exception) as ctx:
+        try:
             raise_exception()
+        except Exception as e:
+            assert e.status_code == status_code
 
-            assert ctx.exception.status_code == status_code
+        with pytest.raises(expected_exception):
+            raise_exception()