Update conftest for unstructuredarchive

icaros-usc · Jun 24, 2024 · 19772cf · 19772cf
1 parent f08d0c9
commit 19772cf
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/archives/conftest.py b/tests/archives/conftest.py
@@ -62,6 +62,7 @@ def add_mode(request):
     "CVTArchive-brute_force",
     "CVTArchive-kd_tree",
     "SlidingBoundariesArchive",
+    "UnstructuredArchive",
 ]
 
 
@@ -146,27 +147,27 @@ def get_archive_data(name, dtype=np.float64):
         archive = UnstructuredArchive(solution_dim=len(solution),
                                       measure_dim=2,
                                       k_neighbors=5,
-                                      sparsity_threshold=1.0,
+                                      novelty_threshold=1.0,
                                       dtype=dtype)
 
         archive_with_elite = UnstructuredArchive(solution_dim=len(solution),
                                                  measure_dim=2,
                                                  k_neighbors=5,
-                                                 sparsity_threshold=1.0,
+                                                 novelty_threshold=1.0,
                                                  dtype=dtype)
         grid_indices = (0,)
         int_index = 0
 
-    archive_with_elite.add_single(solution, objective, measures)  # pylint: disable=E0606
+    archive_with_elite.add_single(solution, objective, measures)
 
     return ArchiveFixtureData(
-        archive,  # pylint: disable=E0606
+        archive,
         archive_with_elite,
         solution,
         objective,
         measures,
         grid_indices,
         int_index,
         centroid,
-        cells,  # pylint: disable=E0606
+        cells,
     )