tests

Genentech · May 14, 2024 · c31ddc4 · c31ddc4
1 parent 7f6a480
commit c31ddc4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/beignet/datasets/test__uniref_dataset.py b/tests/beignet/datasets/test__uniref_dataset.py
@@ -4,8 +4,9 @@
 
 
 class TestUnirefDataset:
-    @mock.patch("beignet.io.download_and_extract_archive", return_value=True)
-    def test_init(self, download_func):
+    @mock.patch("beignet.io._download.download_and_extract_archive", return_value=True)
+    @mock.patch("beignet.datasets.UniRef50Dataset.__init__", return_value=None)
+    def test_init(self, download_func, uniref_init):
         dataset = UniRef50Dataset(root="/tmp/data", download=True)
 
         assert download_func.called_once()