MilesCranmer commited on
Commit
602d0cd
·
1 Parent(s): b77101f

Keep tests the same even with new default

Browse files
Files changed (1) hide show
  1. test/test.py +3 -2
test/test.py CHANGED
@@ -13,6 +13,7 @@ class TestPipeline(unittest.TestCase):
13
  self.default_test_kwargs = dict(
14
  niterations=10,
15
  populations=4,
 
16
  user_input=False,
17
  annealing=True,
18
  useFrequency=False,
@@ -26,10 +27,10 @@ class TestPipeline(unittest.TestCase):
26
  print(equations)
27
  self.assertLessEqual(equations.iloc[-1]["MSE"], 1e-4)
28
 
29
- def test_multiprocessing(self):
30
  y = self.X[:, 0]
31
  equations = pysr(
32
- self.X, y, **self.default_test_kwargs, procs=2, multithreading=False
33
  )
34
  print(equations)
35
  self.assertLessEqual(equations.iloc[-1]["MSE"], 1e-4)
 
13
  self.default_test_kwargs = dict(
14
  niterations=10,
15
  populations=4,
16
+ multithreading=False,
17
  user_input=False,
18
  annealing=True,
19
  useFrequency=False,
 
27
  print(equations)
28
  self.assertLessEqual(equations.iloc[-1]["MSE"], 1e-4)
29
 
30
+ def test_multithreading(self):
31
  y = self.X[:, 0]
32
  equations = pysr(
33
+ self.X, y, **self.default_test_kwargs, procs=2, multithreading=True
34
  )
35
  print(equations)
36
  self.assertLessEqual(equations.iloc[-1]["MSE"], 1e-4)