remove np.round in assert_almost equal

pycomlink · Dec 5, 2023 · e1613f3 · e1613f3
1 parent 19d4a30
commit e1613f3
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pycomlink/tests/test_wet_dry_mlp.py b/pycomlink/tests/test_wet_dry_mlp.py
@@ -3,8 +3,12 @@
 from pycomlink.processing.wet_dry.mlp import mlp_wet_dry
 
 class Testmlppred(unittest.TestCase):
+    """
+    This runs the same tests as test_wet_dry_cnn.py but with different 
+    content in truth_raw.
+    """
+
     def test_mlppred(self):
-        # generate random array
         trsl_channel_1 = np.arange(0, 60 * 8).astype(float)
         trsl_channel_2 = np.arange(0, 60 * 8).astype(float)
 
@@ -63,6 +67,4 @@ def test_mlppred(self):
         )
 
         np.testing.assert_almost_equal(pred[280:293], truth)   
-        np.testing.assert_almost_equal(
-            np.round(pred_raw, decimals=7)[280:293], truth_raw
-        )
+        np.testing.assert_almost_equal(pred_raw[280:293], truth_raw, decimal=7)