catch import error in tests

vda-lab · Jan 3, 2025 · 04b3952 · 04b3952
1 parent 2b89eff
commit 04b3952
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fast_hbcc/tests/test_hbcc.py b/fast_hbcc/tests/test_hbcc.py
@@ -176,7 +176,7 @@ def test_attributes():
         assert isinstance(c.single_linkage_tree_, SingleLinkageTree)
         assert isinstance(c.minimum_spanning_tree_, MinimumSpanningTree)
         assert isinstance(c.approximation_graph_, ApproximationGraph)
-    except ImportError:
+    except ModuleNotFoundError:
         pass
 
 

diff --git a/fast_hbcc/tests/test_sub_clusters.py b/fast_hbcc/tests/test_sub_clusters.py
@@ -68,7 +68,7 @@ def test_attributes():
         assert isinstance(b.condensed_trees_[0], CondensedTree)
         assert isinstance(b.linkage_trees_[0], SingleLinkageTree)
         assert isinstance(b.approximation_graph_, ApproximationGraph)
-    except ImportError:
+    except ModuleNotFoundError:
         pass