Adapt curve fit tests to table locking

biolab · Oct 5, 2021 · f134c10 · f134c10
1 parent dd213c1
commit f134c10
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Orange/regression/tests/test_curvefit.py b/Orange/regression/tests/test_curvefit.py
@@ -12,7 +12,7 @@
 from Orange.preprocess.impute import Random
 from Orange.regression import CurveFitLearner
 from Orange.regression.curvefit import _create_lambda
-
+import Orange.tests
 
 class TestCreateLambda(unittest.TestCase):
     def test_create_lambda_simple(self):
@@ -170,7 +170,8 @@ def test_inadequate_data(self):
 
     def test_missing_values(self):
         data = self.data.copy()
-        data.X[0, 12] = np.nan
+        with data.unlocked():
+            data.X[0, 12] = np.nan
         learner = CurveFitLearner(func, [], ["CRIM"])
         model = learner(data)
         pred = model(data)