fixed tests

dizcza · Oct 10, 2021 · 772caf6 · 772caf6
1 parent 4234340
commit 772caf6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -47,7 +47,7 @@ jobs:
       - run: pip install nose
       - python/save-cache:
           key: pip
-      - run: nosetests --with-coverage --cover-package=mighty mighty/tests
+      - run: nosetests --cover-package=mighty mighty/tests
 
 workflows:
   main:

diff --git a/mighty/tests/test_trainer.py b/mighty/tests/test_trainer.py
@@ -94,10 +94,10 @@ def test_TrainerEmbedding(self):
                                    optimizer=self.optimizer,
                                    scheduler=self.scheduler)
         trainer.env_name = "pytorch-mighty tests"
-        loss_epochs = trainer.train(n_epochs=1, mutual_info_layers=0)
-        assert_array_almost_equal(loss_epochs, [0.63], decimal=2)
-        self.assertEqual(trainer.timer.n_epochs, 1)
-        self.assertEqual(trainer.timer.epoch, 1)
+        loss_epochs = trainer.train(n_epochs=2, mutual_info_layers=0)
+        self.assertAlmostEqual(loss_epochs[-1], 0.38, places=2)
+        self.assertEqual(trainer.timer.n_epochs, 2)
+        self.assertEqual(trainer.timer.epoch, 2)
 
     def test_TrainerEmbedding_cached(self):
         set_seed(3)