From 2da3e7a443d218d172ea135ad7a7c5c1cf054462 Mon Sep 17 00:00:00 2001 From: Donovan Date: Mon, 15 Jul 2024 15:43:06 -0500 Subject: [PATCH] clean up code --- src/nkode_interface.py | 1 - src/pseudo_nkode_api.py | 3 ++- test/test_pseudo_nkode_api.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/nkode_interface.py b/src/nkode_interface.py index a6eb84a..b9b9f61 100644 --- a/src/nkode_interface.py +++ b/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) - diff --git a/src/pseudo_nkode_api.py b/src/pseudo_nkode_api.py index 6ca01da..38628ad 100644 --- a/src/pseudo_nkode_api.py +++ b/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) diff --git a/test/test_pseudo_nkode_api.py b/test/test_pseudo_nkode_api.py index b06bafe..098044f 100644 --- a/test/test_pseudo_nkode_api.py +++ b/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