Spaces:
Sleeping
Sleeping
Commit
·
673c1d2
1
Parent(s):
e790ec3
Move scikit-learn test to miscellaneous tests
Browse files- test/test.py +5 -6
test/test.py
CHANGED
|
@@ -11,12 +11,6 @@ import pandas as pd
|
|
| 11 |
import warnings
|
| 12 |
|
| 13 |
|
| 14 |
-
class TestScikitLearn(unittest.TestCase):
|
| 15 |
-
def test_all_estimators(self):
|
| 16 |
-
model = PySRRegressor()
|
| 17 |
-
check_estimator(model)
|
| 18 |
-
|
| 19 |
-
|
| 20 |
class TestPipeline(unittest.TestCase):
|
| 21 |
def setUp(self):
|
| 22 |
# Using inspect,
|
|
@@ -346,3 +340,8 @@ class TestMiscellaneous(unittest.TestCase):
|
|
| 346 |
with self.assertRaises(Exception) as context:
|
| 347 |
model.fit(X, y)
|
| 348 |
self.assertIn("with 10 features or more", str(context.exception))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
import warnings
|
| 12 |
|
| 13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
class TestPipeline(unittest.TestCase):
|
| 15 |
def setUp(self):
|
| 16 |
# Using inspect,
|
|
|
|
| 340 |
with self.assertRaises(Exception) as context:
|
| 341 |
model.fit(X, y)
|
| 342 |
self.assertIn("with 10 features or more", str(context.exception))
|
| 343 |
+
|
| 344 |
+
def test_scikit_learn_compatibility(self):
|
| 345 |
+
"""Test PySRRegressor compatibility with scikit-learn."""
|
| 346 |
+
model = PySRRegressor()
|
| 347 |
+
check_estimator(model)
|