fix(tests): fd aid algo options

Desbordante · Oct 29, 2024 · c40f71e · c40f71e
1 parent 666826c
commit c40f71e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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),
     )