MilesCranmer commited on
Commit
3663ed6
1 Parent(s): 1a8a8f0

Make sure test doesnt need user input

Browse files
Files changed (2) hide show
  1. .travis.yml +0 -28
  2. test/test.py +6 -3
.travis.yml DELETED
@@ -1,28 +0,0 @@
1
- language: julia
2
- julia:
3
- - 1
4
-
5
- jobs:
6
- include:
7
- - name: "Linux"
8
- os: linux
9
- dist: bionic
10
- before_install: sudo apt-get -y install python3-pip python3-setuptools
11
- env: PY=python3 SETUPPREFIX="--user"
12
- - name: "Windows"
13
- os: windows
14
- before_install:
15
- - choco install python --version 3.8.0
16
- - python -m pip install --upgrade pip
17
- env: PATH=/c/Python38:/c/Python38/Scripts:$PATH PY=python
18
-
19
- install: pip3 install --upgrade pip
20
-
21
- before_script:
22
- - julia --color=yes -e 'using Pkg; pkg"add SymbolicRegression; precompile;"'
23
-
24
- script:
25
- - pip3 install numpy pandas
26
- - $PY setup.py install $SETUPPREFIX
27
- - PATH=$HOME/.local/bin:$PATH $PY test/test.py
28
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
test/test.py CHANGED
@@ -5,7 +5,8 @@ X = np.random.randn(100, 5)
5
  print("Test 1 - defaults; simple linear relation")
6
  y = X[:, 0]
7
  equations = pysr(X, y,
8
- niterations=10)
 
9
  print(equations)
10
  assert equations.iloc[-1]['MSE'] < 1e-4
11
 
@@ -14,7 +15,8 @@ y = X[:, 0]**2
14
  equations = pysr(X, y,
15
  unary_operators=["sq(x) = x^2"], binary_operators=["plus"],
16
  extra_sympy_mappings={'square': lambda x: x**2},
17
- niterations=10)
 
18
  print(equations)
19
  assert equations.iloc[-1]['MSE'] < 1e-4
20
 
@@ -23,7 +25,8 @@ y = X[:, 0] + 3.0
23
  print("Test 3 - empty operator list, and single dimension input")
24
  equations = pysr(X, y,
25
  unary_operators=[], binary_operators=["plus"],
26
- niterations=10)
 
27
 
28
  print(equations)
29
  assert equations.iloc[-1]['MSE'] < 1e-4
 
5
  print("Test 1 - defaults; simple linear relation")
6
  y = X[:, 0]
7
  equations = pysr(X, y,
8
+ niterations=10,
9
+ user_input=False)
10
  print(equations)
11
  assert equations.iloc[-1]['MSE'] < 1e-4
12
 
 
15
  equations = pysr(X, y,
16
  unary_operators=["sq(x) = x^2"], binary_operators=["plus"],
17
  extra_sympy_mappings={'square': lambda x: x**2},
18
+ niterations=10,
19
+ user_input=False)
20
  print(equations)
21
  assert equations.iloc[-1]['MSE'] < 1e-4
22
 
 
25
  print("Test 3 - empty operator list, and single dimension input")
26
  equations = pysr(X, y,
27
  unary_operators=[], binary_operators=["plus"],
28
+ niterations=10,
29
+ user_input=False)
30
 
31
  print(equations)
32
  assert equations.iloc[-1]['MSE'] < 1e-4