Start Python CI15

breimanntools · Apr 5, 2024 · 5bf1a94 · 5bf1a94
1 parent 650bd95
commit 5bf1a94
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 6 deletions.
diff --git a/tests/unit/aaclust_plot_tests/test_aaclust_plot_correlation.py b/tests/unit/aaclust_plot_tests/test_aaclust_plot_correlation.py
@@ -9,8 +9,10 @@
 
 # Helper function
 def adjust_vmin_vmax(vmin=None, vmax=None):
-    vmin = -10000 if vmin < -10000 else vmin
-    vmax = 10000 if vmax > 10000 else vmax
+    if vmin is not None:
+        vmin = -10000 if vmin < -10000 else vmin
+    if vmax is not None:
+        vmax = 10000 if vmax > 10000 else vmax
     return vmin, vmax
 
 

diff --git a/tests/unit/cpp_plot_tests/test_cpp_plot_feature_map.py b/tests/unit/cpp_plot_tests/test_cpp_plot_feature_map.py
@@ -28,8 +28,10 @@
 
 # Helper function
 def adjust_vmin_vmax(vmin=None, vmax=None):
-    vmin = -10000 if vmin < -10000 else vmin
-    vmax = 10000 if vmax > 10000 else vmax
+    if vmin is not None:
+        vmin = -10000 if vmin < -10000 else vmin
+    if vmax is not None:
+        vmax = 10000 if vmax > 10000 else vmax
     return vmin, vmax
 
 

diff --git a/tests/unit/cpp_plot_tests/test_cpp_plot_heatmap.py b/tests/unit/cpp_plot_tests/test_cpp_plot_heatmap.py
@@ -27,9 +27,12 @@
 DICT_COLOR = dict(zip(LIST_CAT, VALID_COLORS))
 
 
+# Helper functions
 def adjust_vmin_vmax(vmin=None, vmax=None):
-    vmin = -10000 if vmin < -10000 else vmin
-    vmax = 10000 if vmax > 10000 else vmax
+    if vmin is not None:
+        vmin = -10000 if vmin < -10000 else vmin
+    if vmax is not None:
+        vmax = 10000 if vmax > 10000 else vmax
     return vmin, vmax