refactor test suite (#56)

e3nn · Dec 24, 2023 · 40f58cf · 40f58cf
1 parent 0b1d0ef
commit 40f58cf
Show file tree

Hide file tree

Showing 35 changed files with 4 additions and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -42,7 +42,7 @@ jobs:
         pip install coveralls
     - name: Test with pytest
       run: |
-        coverage run --source=e3nn_jax -m pytest --doctest-modules --ignore=docs/ --ignore=noxfile.py e3nn_jax examples -x
+        coverage run --source=e3nn_jax -m pytest --doctest-modules --ignore=docs/ --ignore=tests/noxfile.py tests examples
     - name: Upload to coveralls
       if: github.event_name == 'push'
       run: |

diff --git a/pyproject.toml b/pyproject.toml
@@ -89,3 +89,6 @@ exclude_lines = [
 ignore = ["E741", "E203", "W503", "E731", "E721"]
 max-line-length = 127
 max-complexity = 64
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
diff --git a/e3nn_jax/_src/activation_test.py → tests/_src/activation_test.py b/e3nn_jax/_src/activation_test.py → tests/_src/activation_test.py
diff --git a/e3nn_jax/_src/basic_test.py → tests/_src/basic_test.py b/e3nn_jax/_src/basic_test.py → tests/_src/basic_test.py
diff --git a/e3nn_jax/_src/batchnorm/bn_flax_test.py → tests/_src/batchnorm/bn_flax_test.py b/e3nn_jax/_src/batchnorm/bn_flax_test.py → tests/_src/batchnorm/bn_flax_test.py
diff --git a/e3nn_jax/_src/batchnorm/bn_haiku_test.py → tests/_src/batchnorm/bn_haiku_test.py b/e3nn_jax/_src/batchnorm/bn_haiku_test.py → tests/_src/batchnorm/bn_haiku_test.py
diff --git a/e3nn_jax/_src/config_test.py → tests/_src/config_test.py b/e3nn_jax/_src/config_test.py → tests/_src/config_test.py
diff --git a/e3nn_jax/_src/dropout_haiku_test.py → tests/_src/dropout_haiku_test.py b/e3nn_jax/_src/dropout_haiku_test.py → tests/_src/dropout_haiku_test.py
diff --git a/e3nn_jax/_src/gate_test.py → tests/_src/gate_test.py b/e3nn_jax/_src/gate_test.py → tests/_src/gate_test.py
diff --git a/e3nn_jax/_src/grad_test.py → tests/_src/grad_test.py b/e3nn_jax/_src/grad_test.py → tests/_src/grad_test.py
diff --git a/e3nn_jax/_src/irreps_array_test.py → tests/_src/irreps_array_test.py b/e3nn_jax/_src/irreps_array_test.py → tests/_src/irreps_array_test.py
diff --git a/e3nn_jax/_src/irreps_test.py → tests/_src/irreps_test.py b/e3nn_jax/_src/irreps_test.py → tests/_src/irreps_test.py
diff --git a/...x/_src/legacy/core_tensor_product_test.py → ...s/_src/legacy/core_tensor_product_test.py b/...x/_src/legacy/core_tensor_product_test.py → ...s/_src/legacy/core_tensor_product_test.py
diff --git a/e3nn_jax/_src/linear_flax_test.py → tests/_src/linear_flax_test.py b/e3nn_jax/_src/linear_flax_test.py → tests/_src/linear_flax_test.py
diff --git a/e3nn_jax/_src/linear_haiku_test.py → tests/_src/linear_haiku_test.py b/e3nn_jax/_src/linear_haiku_test.py → tests/_src/linear_haiku_test.py
diff --git a/e3nn_jax/_src/perm_test.py → tests/_src/perm_test.py b/e3nn_jax/_src/perm_test.py → tests/_src/perm_test.py
diff --git a/e3nn_jax/_src/radial_test.py → tests/_src/radial_test.py b/e3nn_jax/_src/radial_test.py → tests/_src/radial_test.py
diff --git a/e3nn_jax/_src/radius_graph_test.py → tests/_src/radius_graph_test.py b/e3nn_jax/_src/radius_graph_test.py → tests/_src/radius_graph_test.py
diff --git a/e3nn_jax/_src/reduced_tensor_product_test.py → tests/_src/reduced_tensor_product_test.py b/e3nn_jax/_src/reduced_tensor_product_test.py → tests/_src/reduced_tensor_product_test.py
diff --git a/e3nn_jax/_src/rotation_test.py → tests/_src/rotation_test.py b/e3nn_jax/_src/rotation_test.py → tests/_src/rotation_test.py
diff --git a/e3nn_jax/_src/s2grid_test.py → tests/_src/s2grid_test.py b/e3nn_jax/_src/s2grid_test.py → tests/_src/s2grid_test.py
diff --git a/e3nn_jax/_src/scatter_test.py → tests/_src/scatter_test.py b/e3nn_jax/_src/scatter_test.py → tests/_src/scatter_test.py
diff --git a/e3nn_jax/_src/so3_test.py → tests/_src/so3_test.py b/e3nn_jax/_src/so3_test.py → tests/_src/so3_test.py
diff --git a/e3nn_jax/_src/spherical_harmonics/sh_test.py → tests/_src/spherical_harmonics/sh_test.py b/e3nn_jax/_src/spherical_harmonics/sh_test.py → tests/_src/spherical_harmonics/sh_test.py
diff --git a/...rc/symmetric_tensor_product_haiku_test.py → ...rc/symmetric_tensor_product_haiku_test.py b/...rc/symmetric_tensor_product_haiku_test.py → ...rc/symmetric_tensor_product_haiku_test.py
diff --git a/e3nn_jax/_src/tensor_products_test.py → tests/_src/tensor_products_test.py b/e3nn_jax/_src/tensor_products_test.py → tests/_src/tensor_products_test.py
diff --git a/e3nn_jax/_src/utils/einsum_test.py → tests/_src/utils/einsum_test.py b/e3nn_jax/_src/utils/einsum_test.py → tests/_src/utils/einsum_test.py
diff --git a/e3nn_jax/_src/utils/jit_test.py → tests/_src/utils/jit_test.py b/e3nn_jax/_src/utils/jit_test.py → tests/_src/utils/jit_test.py
diff --git a/conftest.py → tests/conftest.py b/conftest.py → tests/conftest.py
diff --git a/e3nn_jax/experimental/linear_shtp_test.py → tests/experimental/linear_shtp_test.py b/e3nn_jax/experimental/linear_shtp_test.py → tests/experimental/linear_shtp_test.py
diff --git a/...ax/experimental/point_convolution_test.py → tests/experimental/point_convolution_test.py b/...ax/experimental/point_convolution_test.py → tests/experimental/point_convolution_test.py
diff --git a/e3nn_jax/experimental/transformer_test.py → tests/experimental/transformer_test.py b/e3nn_jax/experimental/transformer_test.py → tests/experimental/transformer_test.py
diff --git a/...ax/experimental/voxel_convolution_test.py → tests/experimental/voxel_convolution_test.py b/...ax/experimental/voxel_convolution_test.py → tests/experimental/voxel_convolution_test.py
diff --git a/e3nn_jax/experimental/voxel_pooling_test.py → tests/experimental/voxel_pooling_test.py b/e3nn_jax/experimental/voxel_pooling_test.py → tests/experimental/voxel_pooling_test.py
diff --git a/noxfile.py → tests/noxfile.py b/noxfile.py → tests/noxfile.py