diff --git a/tests/test_repo.py b/tests/test_repo.py index 3b54f099..f7d3ff29 100644 --- a/tests/test_repo.py +++ b/tests/test_repo.py @@ -11,7 +11,7 @@ class TestRepo(unittest.TestCase): """ - Testing interaction with the model repository. + Testing our wrappers around HTRMoPo """ def setUp(self): @@ -33,30 +33,11 @@ def test_get_description(self): Tests fetching the description of a model. """ record = repo.get_description('10.5281/zenodo.8425684') - self.assertEqual(record['doi'], '10.5281/zenodo.8425684') - - def test_get_model(self): - """ - Tests fetching a model. - """ - id = repo.get_model('10.5281/zenodo.8425684', - path=self.temp_model.name) - self.assertEqual(id, 'omnisyr_best.mlmodel') - self.assertEqual((self.temp_path / id).stat().st_size, 16245671) + self.assertEqual(record.doi, '10.5281/zenodo.8425684') def test_prev_record_version_get_description(self): """ Tests fetching the description of a model that has a superseding newer version. """ record = repo.get_description('10.5281/zenodo.6657809') - self.assertEqual(record['doi'], '10.5281/zenodo.6657809') - - def test_prev_record_version_get_model(self): - """ - Tests fetching a model that has a superseding newer version. - """ - id = repo.get_model('10.5281/zenodo.6657809', - path=self.temp_model.name) - self.assertEqual(id, 'HTR-United-Manu_McFrench.mlmodel') - self.assertEqual((self.temp_path / id).stat().st_size, 16176844) - + self.assertEqual(record.doi, '10.5281/zenodo.6657809')