Spaces:
Running
Running
MilesCranmer
commited on
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)
|