basic tests

Author: sydney-runkle

tests/test_errors.py

@@ -551,7 +551,7 @@ def multi_raise_py_error(v: Any) -> Any:
         s2.validate_python('anything')
 
     cause_group = exc_info.value.__cause__
-    assert isinstance(cause_group, BaseExceptionGroup)  # noqa: F821
+    assert isinstance(cause_group, BaseExceptionGroup)
     assert len(cause_group.exceptions) == 1
 
     cause = cause_group.exceptions[0]
@@ -576,7 +576,7 @@ def outer_raise_py_error(v: Any) -> Any:
     with pytest.raises(ValidationError) as exc_info:
         s3.validate_python('anything')
 
-    assert isinstance(exc_info.value.__cause__, BaseExceptionGroup)  # noqa: F821
+    assert isinstance(exc_info.value.__cause__, BaseExceptionGroup)
     assert len(exc_info.value.__cause__.exceptions) == 1
     cause = exc_info.value.__cause__.exceptions[0]
     assert cause.__notes__ and cause.__notes__[-1].startswith('\nPydantic: ')
@@ -585,7 +585,7 @@ def outer_raise_py_error(v: Any) -> Any:
     assert isinstance(subcause, ValidationError)
 
     cause_group = subcause.__cause__
-    assert isinstance(cause_group, BaseExceptionGroup)  # noqa: F821
+    assert isinstance(cause_group, BaseExceptionGroup)
     assert len(cause_group.exceptions) == 1
 
     cause = cause_group.exceptions[0]

tests/validators/test_url.py

@@ -1305,3 +1305,19 @@ def test_url_build() -> None:
     )
     assert url == Url('postgresql://testuser:[email protected]:5432/database?sslmode=require#test')
     assert str(url) == 'postgresql://testuser:[email protected]:5432/database?sslmode=require#test'
+
+
+def test_url_subclass() -> None:
+    class UrlSubclass(Url):
+        pass
+
+    validator = SchemaValidator(core_schema.url_schema(cls=UrlSubclass))
+    assert isinstance(validator.validate_python('http://example.com'), UrlSubclass)
+
+
+def test_multi_host_url_subclass() -> None:
+    class MultiHostUrlSubclass(MultiHostUrl):
+        pass
+
+    validator = SchemaValidator(core_schema.multi_host_url_schema(cls=MultiHostUrlSubclass))
+    assert isinstance(validator.validate_python('http://example.com'), MultiHostUrlSubclass)