Fixed test for grav_index for NaN values

talmolab · Sep 13, 2023 · 28abf45 · 28abf45
1 parent a1a5550
commit 28abf45
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_trait_pipelines.py b/tests/test_trait_pipelines.py
@@ -62,9 +62,9 @@ def test_younger_monocot_pipeline(rice_h5, rice_folder):
         ), f"Unexpected dtype for column {col} in all_traits"
 
     # Value range assertions for traits
-    # assert (
-    #     rice_traits["grav_index"] >= 0
-    # ).all(), "grav_index in rice_traits contains negative values"
+    assert (
+        rice_traits["grav_index"].fillna(0) >= 0
+    ).all(), "grav_index in rice_traits contains negative values"
     assert (
         all_traits["grav_index_median"] >= 0
     ).all(), "grav_index in all_traits contains negative values"