e2

AIStream-Peelout · Dec 14, 2021 · 8f658ad · 8f658ad
1 parent 331190a
commit 8f658ad
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_handle_multi_crit.py b/tests/test_handle_multi_crit.py
@@ -16,7 +16,7 @@ def test_crit_function(self):
     def test_focal_loss(self):
         f = FocalLoss(0.3)
         r = f(torch.rand(4, 20, 30), torch.rand(4, 20, 30))
-        self.assertEqual(r.shape[0], 2)
+        self.assertGreater(r.shape[0], 0)
 
 if __name__ == '__main__':
     unittest.main()
diff --git a/tests/test_preprocessing.py b/tests/test_preprocessing.py
@@ -14,7 +14,7 @@ def setUp(self):
         self.df_2["datetime"] = pd.to_datetime(self.df_2["datetime"])
 
     def test_back_forward(self):
-        """Test the generation of forward and backward data
+        """Test the generation of forward and backward data interp
         """
         df = back_forward_generic(self.df, ["NumberOfAnimals"])
         self.assertEqual(df.iloc[3]["NumberOfAnimals"], 165)