Spaces:
Running
Running
Commit
·
d85c1a5
1
Parent(s):
1858a22
Fix removal of tmp dir in test
Browse files- test/test.py +2 -1
test/test.py
CHANGED
|
@@ -41,7 +41,7 @@ class TestPipeline(unittest.TestCase):
|
|
| 41 |
self.assertLessEqual(equations[0].iloc[-1]["MSE"], 1e-4)
|
| 42 |
self.assertLessEqual(equations[1].iloc[-1]["MSE"], 1e-4)
|
| 43 |
|
| 44 |
-
def
|
| 45 |
y = self.X[:, [0, 1]] ** 2
|
| 46 |
w = np.random.rand(*y.shape)
|
| 47 |
w[w < 0.5] = 0.0
|
|
@@ -61,6 +61,7 @@ class TestPipeline(unittest.TestCase):
|
|
| 61 |
**self.default_test_kwargs,
|
| 62 |
procs=0,
|
| 63 |
temp_equation_file=True,
|
|
|
|
| 64 |
)
|
| 65 |
|
| 66 |
np.testing.assert_almost_equal(
|
|
|
|
| 41 |
self.assertLessEqual(equations[0].iloc[-1]["MSE"], 1e-4)
|
| 42 |
self.assertLessEqual(equations[1].iloc[-1]["MSE"], 1e-4)
|
| 43 |
|
| 44 |
+
def test_multioutput_weighted_with_callable_temp_equation(self):
|
| 45 |
y = self.X[:, [0, 1]] ** 2
|
| 46 |
w = np.random.rand(*y.shape)
|
| 47 |
w[w < 0.5] = 0.0
|
|
|
|
| 61 |
**self.default_test_kwargs,
|
| 62 |
procs=0,
|
| 63 |
temp_equation_file=True,
|
| 64 |
+
delete_tempfiles=False,
|
| 65 |
)
|
| 66 |
|
| 67 |
np.testing.assert_almost_equal(
|