test

Author: bbayukari

pytest/test_workflow.py

@@ -55,8 +55,8 @@ def test_config(model, solver_creator):
     solver = solver_creator(model["n_features"], model["n_informative"])
     solver.set_config(**solver.get_config())
     solver.solve(model["loss"], jit=True)
-
-    assert set(model["support_set"]) == set(solver.support_set)
+    res = solver.get_result()
+    assert set(model["support_set"]) == set(res["support_set"])
 
 
 @pytest.mark.parametrize("model", models, ids=models_ids)