clean up code

src/nkode_interface.py

@@ -45,4 +45,3 @@ class CustomerInterface(BaseModel):
     def get_set_index(self, set_val: int) -> int:
         assert(set_val in self.set_vals)
         return self.set_vals.index(set_val)
-

src/pseudo_nkode_api.py

@@ -50,6 +50,7 @@ class CustomerDBModel(BaseModel):
 
         return False
 
+
 class SessionCacheModel(BaseModel):
     session_id: UUID
     set_interface: list[int] | None = None
@@ -161,7 +162,7 @@ class PseudoNKodeAPI(BaseModel):
         return passcode
 
     def login(self, customer_id: UUID, username: str, key_selection: list[int]) -> bool:
-        assert(customer_id in self.customers.keys())
+        assert (customer_id in self.customers.keys())
         customer = self.customers[customer_id]
         return customer.valid_key_entry(username, key_selection)
 

test/test_pseudo_nkode_api.py

@@ -1,5 +1,5 @@
 import pytest
-from src.pseudo_nkode_api import PseudoNKodeAPI, CustomerDBModel
+from src.pseudo_nkode_api import PseudoNKodeAPI
 
 
 @pytest.fixture()
@@ -31,4 +31,4 @@ def test_create_new_user(pseudo_nkode_api, numb_keys, attrs_per_key, user_passco
     login_interface = pseudo_nkode_api.get_login_index_interface(username, customer.customer_id)
     login_key_selection = key_selection(login_interface)
     successful_login = pseudo_nkode_api.login(customer.customer_id, username, login_key_selection)
-    assert (successful_login)
+    assert successful_login