Skip to content

Commit e70be21

Browse files
committed
feat: test cases
1 parent bf991a4 commit e70be21

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

server/workers/common/common/aquanavi/mapping.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -291,12 +291,15 @@ def load_and_prepare_dataframe():
291291
DataFrame: Pandas DataFrame with information from CSVs.
292292
"""
293293
csv_real_path = Path(CSV_PATH_WITH_REAL_DATA)
294+
csv_test_path = Path(CSV_PATH_WITH_TEST_DATA)
294295

295296
check_that_csv_file_exists(csv_real_path)
297+
check_that_csv_file_exists(csv_test_path)
296298

297299
df_real = pd.read_csv(csv_real_path).fillna("")
300+
df_test = pd.read_csv(csv_test_path).fillna("")
298301

299-
df = pd.concat([df_real], ignore_index=True)
302+
df = pd.concat([df_real, df_test], ignore_index=True)
300303
check_that_required_columns_exists(df, CSV_PATH_WITH_REAL_DATA)
301304

302305
# remove duplicates based on all columns

0 commit comments

Comments
 (0)