summaryrefslogtreecommitdiff
path: root/debian/tests/upstream.py
blob: 5c26cb1eb2deedc68e628f9f1253572867ee480b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import sys
import itertools
import pytest

blacklist = ["test_dtm.py", # Requires hnswlib
             "test_knn.py", # Requires pykeops
             "test_representations.py", # Look into this
             "test_wasserstein_distance.py::test_wasserstein_distance_pot", # Requires eagerpy.
             "test_wasserstein_distance.py::test_wasserstein_distance_grad" # Requires eagerpy.
            ] 

print("Blacklist: %s" %(str(blacklist)))

testdir = sys.argv[1]

deselect_args = list(itertools.chain.from_iterable([["--deselect", "%s/%s" %(testdir, exclude)] for exclude in blacklist]))

exitcode = pytest.main([testdir] + deselect_args)
exit(exitcode)