From 51902423a8465be0bdec314a92ce0031a3a6dd27 Mon Sep 17 00:00:00 2001 From: megalinter-bot <129584137+megalinter-bot@users.noreply.github.com> Date: Wed, 8 May 2024 23:22:57 +0000 Subject: [PATCH] style: apply automated linter fixes --- src/safeds_datasets/image/_mnist/_mnist.py | 4 ++- .../image/_mnist/test_mnist.py | 27 +++++++++++++------ 2 files changed, 22 insertions(+), 9 deletions(-) diff --git a/src/safeds_datasets/image/_mnist/_mnist.py b/src/safeds_datasets/image/_mnist/_mnist.py index 1dc6961..f848453 100644 --- a/src/safeds_datasets/image/_mnist/_mnist.py +++ b/src/safeds_datasets/image/_mnist/_mnist.py @@ -204,7 +204,9 @@ def _load_mnist_like( [labels[label_index] for label_index in array("B", label_file.read())], ) else: - test_labels = Column(file_name, [labels[label_index] for label_index in array("B", label_file.read())]) + test_labels = Column( + file_name, [labels[label_index] for label_index in array("B", label_file.read())], + ) else: with gzip.open(path / file_path, mode="rb") as image_file: magic, size, rows, cols = struct.unpack(">IIII", image_file.read(16)) diff --git a/tests/safeds_datasets/image/_mnist/test_mnist.py b/tests/safeds_datasets/image/_mnist/test_mnist.py index a692053..b526bd8 100644 --- a/tests/safeds_datasets/image/_mnist/test_mnist.py +++ b/tests/safeds_datasets/image/_mnist/test_mnist.py @@ -4,8 +4,7 @@ import pytest from safeds.data.labeled.containers import ImageDataset - -from safeds_datasets.image import load_mnist, _mnist, load_fashion_mnist, load_kmnist +from safeds_datasets.image import _mnist, load_fashion_mnist, load_kmnist, load_mnist class TestMNIST: @@ -22,11 +21,15 @@ def test_should_download_and_return_mnist(self) -> None: assert len(test) == 10_000 train_output = train.get_output() test_output = test.get_output() - assert set(train_output.get_unique_values()) == set(test_output.get_unique_values()) == set(_mnist._mnist._mnist_labels.values()) + 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, pytest.raises(FileNotFoundError): - load_mnist(tmpdirname, download=False) + load_mnist(tmpdirname, download=False) class TestFashionMNIST: @@ -43,11 +46,15 @@ def test_should_download_and_return_mnist(self) -> None: assert len(test) == 10_000 train_output = train.get_output() test_output = test.get_output() - assert set(train_output.get_unique_values()) == set(test_output.get_unique_values()) == set(_mnist._mnist._fashion_mnist_labels.values()) + 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, pytest.raises(FileNotFoundError): - load_fashion_mnist(tmpdirname, download=False) + load_fashion_mnist(tmpdirname, download=False) class TestKMNIST: @@ -64,8 +71,12 @@ def test_should_download_and_return_mnist(self) -> None: assert len(test) == 10_000 train_output = train.get_output() test_output = test.get_output() - assert set(train_output.get_unique_values()) == set(test_output.get_unique_values()) == set(_mnist._mnist._kuzushiji_mnist_labels.values()) + 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, pytest.raises(FileNotFoundError): - load_kmnist(tmpdirname, download=False) + load_kmnist(tmpdirname, download=False)