Create pytest GitHub Action.

PiperOrigin-RevId: 368416101

.github/workflows/ci.yml

@@ -0,0 +1,31 @@
+name: pytest
+
+on: [pull_request]
+
+jobs:
+  test-ubuntu:
+    name: "Python ${{ matrix.python-version }} on ${{ matrix.os }}"
+    runs-on: "${{ matrix.os }}"
+
+    strategy:
+      matrix:
+        python-version: [3.6, 3.7, 3.8, 3.9]
+        os: [ubuntu-latest]
+
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v1
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install -r requirements.txt
+        pip install -r requirements-jax.txt
+        pip install -r requirements-test.txt
+        pip install .
+    - name: Test with pytest
+      run: |
+        pip install pytest
+        pytest jmp

requirements.txt

@@ -1 +1,2 @@
 numpy>=1.19.5
+dataclasses>=0.7 ; python_version<'3.7'