diff --git a/test/test_database.py b/test/test_database.py index c69c18d69f9eb64196df78b9cd43e79a3cb1d2f3..0e55058079cd2ebcd3b4a9c67df5369bc3c335c5 100644 --- a/test/test_database.py +++ b/test/test_database.py @@ -5,7 +5,7 @@ import unittest import json from soitool.database import Database -TESTDATA_PATH = Path(__file__).parent.parent/"soitool/testdata" +TESTDATA_PATH = Path(__file__).parent.parent / "soitool/testdata" class DatabaseTest(unittest.TestCase): @@ -149,9 +149,7 @@ class DatabaseTest(unittest.TestCase): self.assertEqual(entry['code'], actual[i]['code']) def test_get_codebook_small(self): - """ - Assert function get_codebook only return the small codebook. - """ + """Assert function get_codebook only return the small codebook.""" # Load full codebook file_path = os.path.join(TESTDATA_PATH, "codebook.json") with open(file_path, "r") as file: @@ -161,7 +159,7 @@ class DatabaseTest(unittest.TestCase): # Fill expected with only small codebook entries expected = [] for entry in data: - if (entry['type'] == 1): + if entry['type'] == 1: expected.append(entry) # Get small codebook from db