Execute tests with plain python3.

Instead of relying on `pytest`, this change executes the tests directly.

PiperOrigin-RevId: 642317747

.github/workflows/tests.yml

@@ -33,6 +33,8 @@ jobs:
           python setup.py install
           pip install .[testing]
 
-      - name: Test with pytest
+      - name: Run tests
         run: |
-          pytest -n "$(grep -c ^processor /proc/cpuinfo)" android_env
+          # Find all test files, print their names and execute them in parallel
+          # with a maximum of 20 proccesses.
+          find . -type f -name "*_test.py" -print0 | xargs -t -0 -n1 -P 20 python3

pyproject.toml

@@ -30,7 +30,6 @@ testing = [
     "gym",
     "pillow",
     "pytype",
-    "pytest-xdist",
 ]
 acme = ["dm-acme"]
 gym = ["gym"]

setup.py

@@ -48,7 +48,6 @@
     'attrs==20.3.0',  # temporary pin to fix pytype issue.
     'pillow',
     'pytype',
-    'pytest-xdist',
     'gym',
 ]