@@ -42,12 +42,12 @@ def test_find_nodestream_yaml_not_found():
42
42
43
43
44
44
def test_render_property ():
45
- property_name = "name "
45
+ property_name = "id "
46
46
property = PropertyMetadata (is_key = True )
47
47
owner = GraphObjectSchema (name = "Person" )
48
48
node = render_property (owner , property_name , property )
49
49
50
- expected_keys = {"owner" : "Person" , "name " : "name " }
50
+ expected_keys = {"owner" : "Person" , "id " : "id " }
51
51
assert node .key_values == expected_keys
52
52
assert node .properties ["type" ] == property .type .value
53
53
assert node .properties ["is_key" ] == property .is_key
@@ -64,26 +64,26 @@ def test_rel_by_name():
64
64
node = rel_by_name (name )
65
65
assert isinstance (node , Node )
66
66
assert node .type == "RelationshipType"
67
- assert node .key_values ["name " ] == name
67
+ assert node .key_values ["id " ] == name
68
68
69
69
70
70
def test_node_by_name ():
71
71
name = "NODE_NAME"
72
72
node = node_by_name (name )
73
73
assert isinstance (node , Node )
74
74
assert node .type == "NodeType"
75
- assert node .key_values ["name " ] == name
75
+ assert node .key_values ["id " ] == name
76
76
77
77
78
78
def test_render_node ():
79
79
schema = GraphObjectSchema (
80
- name = "Person" , properties = {"name " : PropertyMetadata (is_key = True )}
80
+ name = "Person" , properties = {"id " : PropertyMetadata (is_key = True )}
81
81
)
82
82
ingestion = render_node (schema )
83
83
assert isinstance (ingestion , DesiredIngestion )
84
- assert ingestion .source .key_values ["name " ] == "Person"
84
+ assert ingestion .source .key_values ["id " ] == "Person"
85
85
assert len (ingestion .relationships ) == 1
86
- assert ingestion .relationships [0 ].to_node .key_values ["name " ] == "name "
86
+ assert ingestion .relationships [0 ].to_node .key_values ["id " ] == "id "
87
87
88
88
89
89
def test_render_relationship ():
@@ -92,9 +92,9 @@ def test_render_relationship():
92
92
)
93
93
ingestion = render_relationship (schema )
94
94
assert isinstance (ingestion , DesiredIngestion )
95
- assert ingestion .source .key_values ["name " ] == "FRIENDS_WITH"
95
+ assert ingestion .source .key_values ["id " ] == "FRIENDS_WITH"
96
96
assert len (ingestion .relationships ) == 1
97
- assert ingestion .relationships [0 ].to_node .key_values ["name " ] == "since"
97
+ assert ingestion .relationships [0 ].to_node .key_values ["id " ] == "since"
98
98
99
99
100
100
def test_render_adjacency ():
@@ -105,9 +105,9 @@ def test_render_adjacency():
105
105
assert isinstance (ingestion , DesiredIngestion )
106
106
assert ingestion .source .key_values ["id" ] == "Person_City_LIVES_IN"
107
107
assert len (ingestion .relationships ) == 3
108
- assert ingestion .relationships [0 ].to_node .key_values ["name " ] == "Person"
109
- assert ingestion .relationships [1 ].to_node .key_values ["name " ] == "City"
110
- assert ingestion .relationships [2 ].to_node .key_values ["name " ] == "LIVES_IN"
108
+ assert ingestion .relationships [0 ].to_node .key_values ["id " ] == "Person"
109
+ assert ingestion .relationships [1 ].to_node .key_values ["id " ] == "City"
110
+ assert ingestion .relationships [2 ].to_node .key_values ["id " ] == "LIVES_IN"
111
111
112
112
113
113
def test_schema_renderer_from_file_data ():
@@ -135,7 +135,7 @@ def test_schema_renderer_render_schema():
135
135
schema = Schema ()
136
136
schema .put_node_type (
137
137
GraphObjectSchema (
138
- name = "Person" , properties = {"name " : PropertyMetadata (is_key = True )}
138
+ name = "Person" , properties = {"id " : PropertyMetadata (is_key = True )}
139
139
)
140
140
)
141
141
schema .put_relationship_type (
0 commit comments