Skip to content

Commit 3e7fd05

Browse files
committed
Fix test failures
1 parent 4db84e0 commit 3e7fd05

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

redisgraph_bulk_loader/relation_type.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,6 @@ class RelationType(EntityFile):
1010
def __init__(self, query_buffer, infile, type_str, config):
1111
super(RelationType, self).__init__(infile, type_str, config)
1212
self.query_buffer = query_buffer
13-
self.start_namespace = None
14-
self.end_namespace = None
1513

1614
def process_schemaless_header(self, header):
1715
if self.column_count < 2:
@@ -37,6 +35,8 @@ def post_process_header_with_schema(self, header):
3735

3836
self.start_id = self.types.index(Type.START_ID)
3937
self.end_id = self.types.index(Type.END_ID)
38+
self.start_namespace = None
39+
self.end_namespace = None
4040
# Capture namespaces of start and end IDs if provided
4141
start_match = re.search(r"\((\w+)\)", header[self.start_id])
4242
if start_match:

test/test_bulk_loader.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -583,8 +583,8 @@ def test13_id_namespaces(self):
583583
graph = Graph(graphname, self.redis_con)
584584
query_result = graph.query('MATCH (src)-[]->(dest) RETURN src.id, src.name, LABELS(src), dest.id, dest.views, LABELS(dest) ORDER BY src.id')
585585

586-
expected_result = [[0, 'Jeffrey', 'User', 0, 20, 'Post'],
587-
[1, 'Filipe', 'User', 1, 40, 'Post']]
586+
expected_result = [['0', 'Jeffrey', 'User', '0', 20, 'Post'],
587+
['1', 'Filipe', 'User', '1', 40, 'Post']]
588588
self.assertEqual(query_result.result_set, expected_result)
589589

590590
def test14_array_properties_inferred(self):

0 commit comments

Comments
 (0)