refactor: linter

Safe-DS · May 8, 2024 · 5782725 · 5782725
1 parent 6d897d0
commit 5782725
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tests/safeds_datasets/image/_mnist/test_mnist.py b/tests/safeds_datasets/image/_mnist/test_mnist.py
@@ -25,8 +25,7 @@ def test_should_download_and_return_mnist(self) -> None:
             assert set(train_output.get_unique_values()) == set(test_output.get_unique_values()) == set(_mnist._mnist._mnist_labels.values())
 
     def test_should_raise_if_file_not_found(self) -> None:
-        with tempfile.TemporaryDirectory() as tmpdirname:
-            with pytest.raises(FileNotFoundError):
+        with tempfile.TemporaryDirectory() as tmpdirname, pytest.raises(FileNotFoundError):
                 load_mnist(tmpdirname, download=False)
 
 
@@ -47,8 +46,7 @@ def test_should_download_and_return_mnist(self) -> None:
             assert set(train_output.get_unique_values()) == set(test_output.get_unique_values()) == set(_mnist._mnist._fashion_mnist_labels.values())
 
     def test_should_raise_if_file_not_found(self) -> None:
-        with tempfile.TemporaryDirectory() as tmpdirname:
-            with pytest.raises(FileNotFoundError):
+        with tempfile.TemporaryDirectory() as tmpdirname, pytest.raises(FileNotFoundError):
                 load_fashion_mnist(tmpdirname, download=False)
 
 
@@ -69,6 +67,5 @@ def test_should_download_and_return_mnist(self) -> None:
             assert set(train_output.get_unique_values()) == set(test_output.get_unique_values()) == set(_mnist._mnist._kuzushiji_mnist_labels.values())
 
     def test_should_raise_if_file_not_found(self) -> None:
-        with tempfile.TemporaryDirectory() as tmpdirname:
-            with pytest.raises(FileNotFoundError):
+        with tempfile.TemporaryDirectory() as tmpdirname, pytest.raises(FileNotFoundError):
                 load_kmnist(tmpdirname, download=False)