From c40f71e1cdf53c5bec9d29f1fa65d1fb6357676f Mon Sep 17 00:00:00 2001 From: VladYoSlav Date: Tue, 29 Oct 2024 14:18:47 +0300 Subject: [PATCH] fix(tests): fd aid algo options --- tests/integration/test_retrieve_task_info.py | 2 +- tests/integration/test_set_profiling_task.py | 4 ++-- tests/unit/repository/postgres/test_task.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/integration/test_retrieve_task_info.py b/tests/integration/test_retrieve_task_info.py index 0b0be869..77385169 100644 --- a/tests/integration/test_retrieve_task_info.py +++ b/tests/integration/test_retrieve_task_info.py @@ -16,7 +16,7 @@ def test_retrieve_task(client: TestClient, context): assert dataset_response.status_code == 200 dataset_id = UUID(dataset_response.json()) - algo_config = AidConfig(algo_name=FdAlgoName.Aid, is_null_equal_null=True) + algo_config = AidConfig(algo_name=FdAlgoName.Aid, max_lhs=1) task_config = FdTaskConfig(primitive_name=PrimitiveName.fd, config=algo_config) task_response = set_task(client, dataset_id, task_config) diff --git a/tests/integration/test_set_profiling_task.py b/tests/integration/test_set_profiling_task.py index 568121a8..87df3449 100644 --- a/tests/integration/test_set_profiling_task.py +++ b/tests/integration/test_set_profiling_task.py @@ -15,7 +15,7 @@ def test_set_profiling_task(client: TestClient, context): assert dataset_response.status_code == 200 dataset_id = UUID(dataset_response.json()) - algo_config = AidConfig(algo_name=FdAlgoName.Aid, is_null_equal_null=True) + algo_config = AidConfig(algo_name=FdAlgoName.Aid, max_lhs=1) task_config = FdTaskConfig(primitive_name=PrimitiveName.fd, config=algo_config) response = set_task(client, dataset_id, task_config) @@ -33,7 +33,7 @@ def test_set_profiling_task(client: TestClient, context): def test_set_profiling_task_with_non_existent_dataset(client: TestClient): dataset_id = uuid4() - algo_config = AidConfig(algo_name=FdAlgoName.Aid, is_null_equal_null=True) + algo_config = AidConfig(algo_name=FdAlgoName.Aid, max_lhs=1) task_config = FdTaskConfig(primitive_name=PrimitiveName.fd, config=algo_config) response = set_task(client, dataset_id, task_config) diff --git a/tests/unit/repository/postgres/test_task.py b/tests/unit/repository/postgres/test_task.py index 0ac36a21..23413686 100644 --- a/tests/unit/repository/postgres/test_task.py +++ b/tests/unit/repository/postgres/test_task.py @@ -52,7 +52,7 @@ def dataset_id(dataset_create_schema, postgres_context): def get_config(): return FdTaskConfig( primitive_name=PrimitiveName.fd, - config=AidConfig(algo_name=FdAlgoName.Aid, is_null_equal_null=True), + config=AidConfig(algo_name=FdAlgoName.Aid, max_lhs=1), )