From f00e07ad1c61f3f1ee71892daf4c25ffc40a9311 Mon Sep 17 00:00:00 2001 From: Leon Luttenberger Date: Mon, 18 Dec 2023 15:01:37 -0600 Subject: [PATCH] fix neptune tests --- tests/unit/test_neptune_parsing.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/tests/unit/test_neptune_parsing.py b/tests/unit/test_neptune_parsing.py index fbb6b9e58..9f4d8bbad 100644 --- a/tests/unit/test_neptune_parsing.py +++ b/tests/unit/test_neptune_parsing.py @@ -51,11 +51,12 @@ 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") @@ -63,11 +64,12 @@ 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[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"}