Deflaked file

ocbe-uio · Jan 14, 2025 · 9d76b94 · 9d76b94
1 parent 3495f29
commit 9d76b94
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py
@@ -95,6 +95,7 @@ def test_normalize_data():
 def test_combine_fractions_shape():
     assert data_shap.shape == (35928, 3)
 
+
 # Fit lines for Discovery and Validation cohorts
 fit_discovery = pipeline.fit_line(data_shap)
 
@@ -106,13 +107,14 @@ def test_fit_line():
     assert 'CI' in fit_discovery.columns
     if local:
         tol = 1e-2
+        line = fit_discovery.loc[fit_discovery['Feature'] == 'B.cells.Memory']
         assert math.isclose(
-            fit_discovery.loc[fit_discovery['Feature'] == 'B.cells.Memory', 'Coef'].values[0],
+            fit_discovery.loc[line, 'Coef'].values[0],
             -0.0199,
             rel_tol=tol
         )
         assert math.isclose(
-            fit_discovery.loc[fit_discovery['Feature'] == 'B.cells.Memory', 'CI'].values[0],
+            fit_discovery.loc[line, 'CI'].values[0],
             0.0000193,  # Replace with actual value from terminal
             rel_tol=tol
         )