Add .travis.yml

.travis.yml

@@ -0,0 +1,11 @@
+language: julia
+os: linux
+dist: bionic
+
+addons:
+    apt:
+        packages:
+            - python3-pip
+
+script:
+    - ./test/travis.sh

setup.py

@@ -12,6 +12,10 @@ setuptools.setup(
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/MilesCranmer/pysr",
+    install_requires=[
+        "numpy",
+        "pandas"
+        ],
     packages=setuptools.find_packages(),
     package_data={
         'pysr': ['../julia/*.jl']

test/test.py

@@ -0,0 +1,7 @@
+import numpy as np
+from pysr import pysr
+X = np.random.randn(100, 5)
+y = X[:, 0]
+
+equations = pysr(X, y)
+print(equations)

test/travis.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+julia -e 'import Pkg; Pkg.add("Optim"); Pkg.add("SpecialFunctions")'
+sudo python3 -m pip install numpy pandas
+sudo python3 setup.py install
+python3 test/test.py
+