From c5cf0194be2cc0e41774e5355cbb53d836306a23 Mon Sep 17 00:00:00 2001 From: kfilippopolitis Date: Tue, 25 Jun 2024 15:15:52 +0300 Subject: [PATCH] fix on test --- mipdb/usecases.py | 4 +++- tests/test_commands.py | 12 +++--------- 2 files changed, 6 insertions(+), 10 deletions(-) diff --git a/mipdb/usecases.py b/mipdb/usecases.py index 0fd1054..93a6f3d 100644 --- a/mipdb/usecases.py +++ b/mipdb/usecases.py @@ -774,7 +774,9 @@ def execute(self) -> None: data_model_table = DataModelTable() dataset_table = DatasetsTable() dataset_row_columns = ["dataset_id", "data_model_id", "code", "label", "status"] - dataset_rows = dataset_table.get_datasets(self.sqlite_db, columns=dataset_row_columns) + dataset_rows = dataset_table.get_datasets( + self.sqlite_db, columns=dataset_row_columns + ) data_model_fullname_by_data_model_id = { data_model_id: get_data_model_fullname(code, version) for data_model_id, code, version in data_model_table.get_data_models( diff --git a/tests/test_commands.py b/tests/test_commands.py index 75ba08b..58e8d56 100644 --- a/tests/test_commands.py +++ b/tests/test_commands.py @@ -334,16 +334,10 @@ def test_invalid_dataset_error_cases(data_model, dataset, exception_message): def test_validate_no_db(): runner = CliRunner() - validation_result = runner.invoke( - validate_folder, - [ - ABSOLUTE_PATH_FAIL_DATA_FOLDER - ] - ) + validation_result = runner.invoke(validate_folder, [ABSOLUTE_PATH_FAIL_DATA_FOLDER]) assert ( - """An error occurred while validating the csv on column: 'var2' - index failure_case -0 0 l3""" in validation_result.stdout + "An error occurred while validating the csv on column: 'var2'" + in validation_result.stdout )