Ignore DataFrame index in e2e assertion

thoughtworks · Jan 4, 2023 · 97fc247 · 97fc247
1 parent af0532f
commit 97fc247
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/e2e/assert.py b/e2e/assert.py
@@ -10,8 +10,8 @@ def standalone_should_match_snapshot():
 
 
 def spark_should_match_snapshot():
-    expect = pd.read_csv("./e2e/expected.csv")
-    actual = dd.read_csv("./e2e/output/spark/*.csv").compute()
+    expect = pd.read_csv("./e2e/expected.csv").reset_index(drop=True)
+    actual = dd.read_csv("./e2e/output/spark/*.csv").compute().reset_index(drop=True)
     pd.testing.assert_frame_equal(expect, actual)
     print("E2E - Spark test passed")