fix neptune tests

tests/unit/test_neptune_parsing.py

@@ -51,23 +51,25 @@ def test_parse_gremlin_edge_elements(gremlin_parser):
     out = gremlin_parser.gremlin_results_to_dict(input)
     df = pd.DataFrame.from_records(out)
     row = df.iloc[0]
-    assert df.shape == (1, 4)
+    assert df.shape == (1, 5)
     assert row["id"] == "foo"
     assert row["outV"] == "out1"
     assert row["label"] == "label"
     assert row["inV"] == "in1"
+    assert row["properties"] is None
 
     # parse multiple edge elements
     v1 = Edge("bar", "out1", "label", "in2")
     input = [v, v1]
     out = gremlin_parser.gremlin_results_to_dict(input)
     df = pd.DataFrame.from_records(out)
     row = df.iloc[1]
-    assert df.shape == (2, 4)
+    assert df.shape == (2, 5)
     assert row["id"] == "bar"
     assert row["outV"] == "out1"
     assert row["label"] == "label"
     assert row["inV"] == "in2"
+    assert row["properties"] is None
 
 
 # parse Property elements
@@ -91,11 +93,10 @@ def test_parse_gremlin_property_elements(gremlin_parser):
     out = gremlin_parser.gremlin_results_to_dict(input)
     df = pd.DataFrame.from_records(out)
     row = df.iloc[0]
-    assert df.shape == (1, 4)
+    assert df.shape == (1, 3)
     assert row["element"] == "bar"
     assert row["value"] == "name"
     assert row["key"] == "foo"
-    assert row["properties"] is None
 
 
 # parse Path elements
@@ -133,7 +134,7 @@ def test_parse_gremlin_path_elements(gremlin_parser):
     out = gremlin_parser.gremlin_results_to_dict([p])
     df = pd.DataFrame.from_records(out)
     row = df.iloc[0]
-    assert df.shape == (1, 4)
+    assert df.shape == (1, 3)
     assert row[0]["name"] == "foo"
     assert row[0]["age"] == 29
     assert row[1]["dist"] == 32
@@ -148,7 +149,7 @@ def test_parse_gremlin_path_elements(gremlin_parser):
     out = gremlin_parser.gremlin_results_to_dict([p])
     df = pd.DataFrame.from_records(out)
     row = df.iloc[0]
-    assert df.shape == (1, 4)
+    assert df.shape == (1, 3)
     assert row[0]["name"] == "foo"
     assert row[0]["age"] == 29
     assert row[1] == {"id": "bar", "label": "label", "outV": "out1", "inV": "in2"}