Skip to content

Commit 296335e

Browse files
committed
feat: test cases
1 parent 40dcef3 commit 296335e

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
@@ -318,12 +318,15 @@ def load_and_prepare_dataframe():
318318
DataFrame: Pandas DataFrame with information from CSVs.
319319
"""
320320
csv_real_path = Path(CSV_PATH_WITH_REAL_DATA)
321+
csv_test_path = Path(CSV_PATH_WITH_TEST_DATA)
321322

322323
check_that_csv_file_exists(csv_real_path)
324+
check_that_csv_file_exists(csv_test_path)
323325

324326
df_real = pd.read_csv(csv_real_path).fillna("")
327+
df_test = pd.read_csv(csv_test_path).fillna("")
325328

326-
df = pd.concat([df_real], ignore_index=True)
329+
df = pd.concat([df_real, df_test], ignore_index=True)
327330
check_that_required_columns_exists(df, CSV_PATH_WITH_REAL_DATA)
328331

329332
# remove duplicates based on all columns

0 commit comments

Comments
 (0)