Fix pydantic v2 tests

rmasters · rmasters · commit 9a0c172c14da · 2023-12-31T16:48:19.000Z
diff --git a/tests/test_advanced/test_column_types/test_tutorial002.py b/tests/test_advanced/test_column_types/test_tutorial002.py
@@ -3,13 +3,14 @@
 import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import get_testing_print_function, needs_pydanticv2
 
 expected_calls = [
     ["Average score:", pytest.approx(0.5079, abs=0.0001)],
 ]
 
 
+@needs_pydanticv2
 def test_tutorial(clear_sqlmodel):
     from docs_src.advanced.column_types import tutorial002 as mod
 
diff --git a/tests/test_advanced/test_column_types/test_tutorial003.py b/tests/test_advanced/test_column_types/test_tutorial003.py
@@ -2,7 +2,7 @@
 
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import get_testing_print_function, needs_pydanticv2
 
 expected_calls = [
     [
@@ -19,6 +19,7 @@
 ]
 
 
+@needs_pydanticv2
 def test_tutorial(clear_sqlmodel):
     from docs_src.advanced.column_types import tutorial003 as mod
 

Original file line number	Diff line number	Diff line change
`@@ -2,7 +2,7 @@`
`2`	`2`
`3`	`3`	`from sqlmodel import create_engine`
`4`	`4`
`5`		`-from ...conftest import get_testing_print_function`
	`5`	`+from ...conftest import get_testing_print_function, needs_pydanticv2`
`6`	`6`
`7`	`7`	`expected_calls = [`
`8`	`8`	`[`
`@@ -19,6 +19,7 @@`
`19`	`19`	`]`
`20`	`20`
`21`	`21`
	`22`	`+@needs_pydanticv2`
`22`	`23`	`def test_tutorial(clear_sqlmodel):`
`23`	`24`	`from docs_src.advanced.column_types import tutorial003 as mod`
`24`	`25`