Now even more beautiful code

data_adapter_oemof/build_datapackage.py

@@ -4,7 +4,6 @@
 from typing import Optional
 
 import pandas as pd
-import numpy as np
 from data_adapter import core
 from data_adapter.preprocessing import Adapter
 from datapackage import Package

data_adapter_oemof/mappings.py

@@ -115,8 +115,10 @@ def get_data(self, key, field_type: Optional[Type] = None):
                     f"Using existing timeseries column '{timeseries_key}'."
                 )
                 return timeseries_key
-            logger.info(f"For Process {self.process_name} no timeseries found "
-                        f"entry for mapped key '{key}'")
+            logger.info(
+                f"For Process {self.process_name} no timeseries found "
+                f"entry for mapped key '{key}'"
+            )
             return None
 
         # 2 Use defaults

tests/test_build_datapackage.py

@@ -6,12 +6,12 @@
 import pytest
 from data_adapter.databus import download_collection
 from data_adapter.preprocessing import Adapter
+from oemof.tabular.datapackage import EnergySystem, Model
+from oemof.tabular.facades import Bus, Commodity, Dispatchable, Load, Storage, Volatile
 from pandas import Timestamp
 from utils import PATH_TEST_FILES, check_if_csv_dirs_equal
 
 from data_adapter_oemof.build_datapackage import DataPackage, refactor_timeseries
-from oemof.tabular.datapackage import EnergySystem, Model
-from oemof.tabular.facades import Dispatchable, Load, Commodity, Storage, Bus, Volatile
 
 path_default = PATH_TEST_FILES / "_files"
 
@@ -290,10 +290,12 @@ def test_read_datapackage():
             "bus": Bus,
             "storage": Storage,
             "volatile": Volatile,
-            "commodity": Commodity
+            "commodity": Commodity,
         },
     )
     model = Model(es)
+    test_model = 1
+    assert model, test_model
 
 
 def test_period_csv_creation():