Spaces:
				
			
			
	
			
			
					
		Running
		
	
	
	
			
			
	
	
	
	
		
		
					
		Running
		
	Commit 
							
							·
						
						775c667
	
1
								Parent(s):
							
							bad0567
								
More descriptive name for warm_start test
Browse files- test/test.py +3 -2
    	
        test/test.py
    CHANGED
    
    | @@ -118,7 +118,7 @@ class TestPipeline(unittest.TestCase): | |
| 118 | 
             
                        print("Model equations: ", model.sympy()[1])
         | 
| 119 | 
             
                        print("True equation: x1^2")
         | 
| 120 |  | 
| 121 | 
            -
                def  | 
| 122 | 
             
                    X = self.rstate.randn(100, 1)
         | 
| 123 | 
             
                    y = X[:, 0] + 3.0
         | 
| 124 | 
             
                    regressor = PySRRegressor(
         | 
| @@ -135,7 +135,8 @@ class TestPipeline(unittest.TestCase): | |
| 135 | 
             
                    np.testing.assert_almost_equal(regressor.predict(X), y, decimal=1)
         | 
| 136 |  | 
| 137 | 
             
                    # Test if repeated fit works:
         | 
| 138 | 
            -
                    regressor.set_params(niterations=0)
         | 
|  | |
| 139 | 
             
                    regressor.fit(X, y)
         | 
| 140 |  | 
| 141 | 
             
                    self.assertLessEqual(regressor.equations_.iloc[-1]["loss"], 1e-4)
         | 
|  | |
| 118 | 
             
                        print("Model equations: ", model.sympy()[1])
         | 
| 119 | 
             
                        print("True equation: x1^2")
         | 
| 120 |  | 
| 121 | 
            +
                def test_empty_operators_single_input_warm_start(self):
         | 
| 122 | 
             
                    X = self.rstate.randn(100, 1)
         | 
| 123 | 
             
                    y = X[:, 0] + 3.0
         | 
| 124 | 
             
                    regressor = PySRRegressor(
         | 
|  | |
| 135 | 
             
                    np.testing.assert_almost_equal(regressor.predict(X), y, decimal=1)
         | 
| 136 |  | 
| 137 | 
             
                    # Test if repeated fit works:
         | 
| 138 | 
            +
                    regressor.set_params(niterations=0, warm_start=True)
         | 
| 139 | 
            +
                    # This should exit immediately, and use the old equations
         | 
| 140 | 
             
                    regressor.fit(X, y)
         | 
| 141 |  | 
| 142 | 
             
                    self.assertLessEqual(regressor.equations_.iloc[-1]["loss"], 1e-4)
         |