Fix test failures (#48)

Author: jeffreylovitz

redisgraph_bulk_loader/relation_type.py

@@ -10,8 +10,6 @@ class RelationType(EntityFile):
     def __init__(self, query_buffer, infile, type_str, config):
         super(RelationType, self).__init__(infile, type_str, config)
         self.query_buffer = query_buffer
-        self.start_namespace = None
-        self.end_namespace = None
 
     def process_schemaless_header(self, header):
         if self.column_count < 2:
@@ -37,6 +35,8 @@ def post_process_header_with_schema(self, header):
 
         self.start_id = self.types.index(Type.START_ID)
         self.end_id = self.types.index(Type.END_ID)
+        self.start_namespace = None
+        self.end_namespace = None
         # Capture namespaces of start and end IDs if provided
         start_match = re.search(r"\((\w+)\)", header[self.start_id])
         if start_match:

test/test_bulk_loader.py

@@ -583,8 +583,8 @@ def test13_id_namespaces(self):
         graph = Graph(graphname, self.redis_con)
         query_result = graph.query('MATCH (src)-[]->(dest) RETURN src.id, src.name, LABELS(src), dest.id, dest.views, LABELS(dest) ORDER BY src.id')
 
-        expected_result = [[0, 'Jeffrey', 'User', 0, 20, 'Post'],
-                           [1, 'Filipe', 'User', 1, 40, 'Post']]
+        expected_result = [['0', 'Jeffrey', 'User', '0', 20, 'Post'],
+                           ['1', 'Filipe', 'User', '1', 40, 'Post']]
         self.assertEqual(query_result.result_set, expected_result)
 
     def test14_array_properties_inferred(self):