test: Updated assert in test_hub (#1495)

Updated all raw asserts to corresponding unittest.TestCase.assert. See #1483

test: Updated assert in test_hub (#1495)
Updated all raw asserts to corresponding unittest.TestCase.assert. See #1483
8bc4ab06 · F-G Fernandez · Francisco Massa · c00222cf · 8bc4ab06
Commit 8bc4ab06 authored Oct 18, 2019 by F-G Fernandez Committed by Francisco Massa Oct 18, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test/test_hub.py test/test_hub.py +1 -1

No files found.
--- a/test/test_hub.py
+++ b/test/test_hub.py
@@ -44,7 +44,7 @@ class TestHub(unittest.TestCase):
            progress=False)
        self.assertEqual(sum_of_model_parameters(hub_model).item(),
                         SUM_OF_PRETRAINED_RESNET18_PARAMS)
-        assert os.path.exists(temp_dir + '/pytorch_vision_master')
+        self.assertTrue(os.path.exists(temp_dir + '/pytorch_vision_master'))
        shutil.rmtree(temp_dir + '/pytorch_vision_master')
    def test_list_entrypoints(self):