blob: 63a35e0bc9883c36a5686c12d4648604ba6a90de (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
import sys
import itertools
import pytest
blacklist = ["test_knn.py", # Requires pykeops
"test_representations.py", # Look into this
"test_wasserstein_distance.py::test_wasserstein_distance_grad" # Requires PyTorch.
]
print("Blacklist: %s" %(str(blacklist)))
# These should be completely ignored, not even loaded (their imports fail).
ignorelist = ["test_dtm.py", # Requires hnswlib.
"test_wasserstein_with_tensors.py" # Requires PyTorch and TF
]
print("Ignorelist:", ignorelist)
testdir = sys.argv[1]
deselect_args = list(itertools.chain.from_iterable([["--deselect", "%s/%s" %(testdir, exclude)] for exclude in blacklist]))
ignore_args = list(itertools.chain.from_iterable([["--ignore-glob", "%s/%s" %(testdir, exclude)] for exclude in ignorelist]))
exitcode = pytest.main([testdir] + deselect_args + ignore_args)
exit(exitcode)
|
