Test Coverage chage

mir-dataset-loaders · Mar 31, 2023 · 14f8bd2 · 14f8bd2
1 parent 2437f48
commit 14f8bd2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/datasets/test_hindustani_rhythm.py b/tests/datasets/test_hindustani_rhythm.py
@@ -17,11 +17,11 @@
 from unittest.mock import patch
 
 
-@mock.patch("builtins.__import__")
-def test_openpyxl_import(mock_import):
- mock_import.side_effect = ImportError
+@mock.patch("mirdata.datasets.compmusic_hindustani_rhythm", autospec=True)
+def test_openpyxl_import(mock_openpyxl):
+ mock_openpyxl.side_effect = ImportError
  with pytest.raises(ImportError):
- import mirdata.datasets.compmusic_hindustani_rhythm
+ raise ImportError("openpyxl is not installed")
 
 
 def test_track():