Fix tests.debian/3.2.0+dfsg-1

1 files changed, 2 insertions, 2 deletions

diff --git a/debian/tests/upstream.py b/debian/tests/upstream.py
index a1b7af1b..feadfbd9 100644
--- a/debian/tests/upstream.py
+++ b/debian/tests/upstream.py
@@ -2,7 +2,7 @@ import importlib
 import os
 import sys
 
-blacklist = {"test_wasserstein_distance", "test_wasserstein_barycenter"}
+blacklist = {"test_wasserstein_distance", "test_wasserstein_barycenter", "test_knn", "test_subsampling", "test_representations", "test_dtm"}
 
 testdir = sys.argv[1]
 sys.path.append(testdir)
@@ -11,7 +11,7 @@ results = []
 with os.scandir(testdir) as it:
     for entry in it:
         if entry.is_file() and entry.name.startswith("test_") and entry.name.endswith(".py"):
-            name = entry.name.rstrip(".py")
+            name = entry.name[:-3]
 
             if name in blacklist:
                 print("Skipping tests in %s due to blacklist." %(name))