read index column automatically

pyiron · Aug 9, 2024 · 53cff6f · 53cff6f
1 parent 4c2aed7
commit 53cff6f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/unit/archiving/test_export.py b/tests/unit/archiving/test_export.py
@@ -40,8 +40,7 @@ def test_exportedCSV(self):
         # and is compared with the return dataframe from export_database
         directory_to_transfer = os.path.basename(self.pr.path[:-1])
         self.pr.pack(destination_path=self.arch_dir, compress=False)
-        df_read = pd.read_csv("export.csv")
-        df_read.drop(df_read.keys()[0], inplace=True, axis=1)
+        df_read = pd.read_csv("export.csv", index_col=0)
         # this removes the "None/NaN/empty" cells as well as the unnamed column
         df_read.dropna(inplace=True, axis=1)
         df_read["timestart"] = pd.to_datetime(df_read["timestart"])