test: save to temp file

MilesCranmer · MilesCranmer · commit fa272c6a6637 · 2024-08-23T14:59:33.000+08:00
diff --git a/pytest/integration/test_pysr.py b/pytest/integration/test_pysr.py
@@ -11,6 +11,8 @@ def test_integration_pysr():
         unary_operators=["cos"],
         binary_operators=["*", "+", "-"],
         early_stop_condition=1e-5,
+        temp_equation_file=True,
+        progress=False,
     )
     model.fit(X, y)
     assert model.equations_.iloc[-1]["loss"] < 1e-5

Original file line number	Diff line number	Diff line change
`@@ -11,6 +11,8 @@ def test_integration_pysr():`
`11`	`11`	`unary_operators=["cos"],`
`12`	`12`	`binary_operators=["*", "+", "-"],`
`13`	`13`	`early_stop_condition=1e-5,`
	`14`	`+ temp_equation_file=True,`
	`15`	`+ progress=False,`
`14`	`16`	`)`
`15`	`17`	`model.fit(X, y)`
`16`	`18`	`assert model.equations_.iloc[-1]["loss"] < 1e-5`