MilesCranmer commited on
Commit
673c1d2
1 Parent(s): e790ec3

Move scikit-learn test to miscellaneous tests

Browse files
Files changed (1) hide show
  1. 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)