Skip to content

Commit 021b5d9

Browse files
author
matmoncon
committed
test: fix test case since pydantic generates different schemas depending on python version
1 parent 2ba47b9 commit 021b5d9

File tree

2 files changed

+3
-28
lines changed

2 files changed

+3
-28
lines changed

pyneo4j_ogm/fields/relationship_property.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -286,9 +286,9 @@ def __ne__(self, other: Any) -> bool:
286286

287287
def __repr__(self) -> str:
288288
return (
289-
f"{self.__class__.__name__}(target_model_name={self._target_model_name} "
290-
f"relationship_model={self._relationship_model_name} direction={self._direction} "
291-
f"cardinality={self._cardinality} allow_multiple={self._allow_multiple})"
289+
f"{self.__class__.__name__}(target_model_name={self._target_model_name}, "
290+
f"relationship_model={self._relationship_model_name}, direction={self._direction}, "
291+
f"cardinality={self._cardinality}, allow_multiple={self._allow_multiple})"
292292
)
293293

294294
def __str__(self) -> str:

tests/unit/test_schema.py

Lines changed: 0 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
from pyneo4j_ogm.pydantic_utils import IS_PYDANTIC_V2, get_schema
44
from tests.fixtures.db_setup import Coffee, Consumed, Developer
5-
from tests.utils.string_utils import assert_string_equality
65

76

87
def test_schema():
@@ -15,12 +14,6 @@ def test_schema():
1514
if IS_PYDANTIC_V2:
1615
assert "coffee" in schema["properties"]
1716
assert "default" in schema["properties"]["coffee"]
18-
assert_string_equality(
19-
schema["properties"]["coffee"]["default"],
20-
"""RelationshipProperty(target_model_name=Coffee relationship_model=Consumed
21-
direction=RelationshipPropertyDirection.OUTGOING cardinality=RelationshipPropertyCardinality.ZERO_OR_MORE
22-
allow_multiple=False)""",
23-
)
2417
assert schema["properties"]["coffee"]["type"] == "object"
2518
assert schema["properties"]["coffee"]["title"] == "Coffee"
2619
assert schema["properties"]["coffee"]["required"] == [
@@ -36,12 +29,6 @@ def test_schema():
3629

3730
assert "colleagues" in schema["properties"]
3831
assert "default" in schema["properties"]["colleagues"]
39-
assert_string_equality(
40-
schema["properties"]["colleagues"]["default"],
41-
"""RelationshipProperty(target_model_name=Developer relationship_model=WorkedWith
42-
direction=RelationshipPropertyDirection.OUTGOING cardinality=RelationshipPropertyCardinality.ZERO_OR_MORE
43-
allow_multiple=True)""",
44-
)
4532
assert schema["properties"]["colleagues"]["type"] == "object"
4633
assert schema["properties"]["colleagues"]["title"] == "Colleagues"
4734
assert schema["properties"]["colleagues"]["required"] == [
@@ -57,12 +44,6 @@ def test_schema():
5744
else:
5845
assert "coffee" in schema["definitions"]["Developer"]["properties"]
5946
assert "default" in schema["definitions"]["Developer"]["properties"]["coffee"]
60-
assert_string_equality(
61-
schema["definitions"]["Developer"]["properties"]["coffee"]["default"],
62-
"""RelationshipProperty(target_model_name=Coffee relationship_model=Consumed
63-
direction=RelationshipPropertyDirection.OUTGOING cardinality=RelationshipPropertyCardinality.ZERO_OR_MORE
64-
allow_multiple=False)""",
65-
)
6647
assert schema["definitions"]["Developer"]["properties"]["coffee"]["type"] == "object"
6748
assert schema["definitions"]["Developer"]["properties"]["coffee"]["title"] == "Coffee"
6849
assert schema["definitions"]["Developer"]["properties"]["coffee"]["required"] == [
@@ -89,12 +70,6 @@ def test_schema():
8970

9071
assert "colleagues" in schema["definitions"]["Developer"]["properties"]
9172
assert "default" in schema["definitions"]["Developer"]["properties"]["colleagues"]
92-
assert_string_equality(
93-
schema["definitions"]["Developer"]["properties"]["colleagues"]["default"],
94-
"""RelationshipProperty(target_model_name=Developer relationship_model=WorkedWith
95-
direction=RelationshipPropertyDirection.OUTGOING cardinality=RelationshipPropertyCardinality.ZERO_OR_MORE
96-
allow_multiple=True)""",
97-
)
9873
assert schema["definitions"]["Developer"]["properties"]["colleagues"]["type"] == "object"
9974
assert schema["definitions"]["Developer"]["properties"]["colleagues"]["title"] == "Colleagues"
10075
assert schema["definitions"]["Developer"]["properties"]["colleagues"]["required"] == [

0 commit comments

Comments
 (0)