From 1682c012b401b1970332815ac5b16ab6d22d326b Mon Sep 17 00:00:00 2001 From: "Charles T. Gray" Date: Sun, 12 Oct 2025 07:52:22 +0200 Subject: [PATCH] close #17 needed to refactor to taking button dat to composition --- button_2/classes/entities/employee.py | 2 +- button_2/tests/entities/test_employee.py | 10 ++-------- 2 files changed, 3 insertions(+), 9 deletions(-) diff --git a/button_2/classes/entities/employee.py b/button_2/classes/entities/employee.py index e58bfe2..7b18c83 100644 --- a/button_2/classes/entities/employee.py +++ b/button_2/classes/entities/employee.py @@ -1,5 +1,5 @@ import random -from classes.data.button_dat import ButtonDat +from ..data.button_dat import ButtonDat class Employee: def __init__(self, button_dat: ButtonDat): diff --git a/button_2/tests/entities/test_employee.py b/button_2/tests/entities/test_employee.py index 94f268f..cc30ad0 100644 --- a/button_2/tests/entities/test_employee.py +++ b/button_2/tests/entities/test_employee.py @@ -3,10 +3,7 @@ from button_2.classes.data.button_dat import ButtonDat button_dat = ButtonDat() -employee = Employee( - job_title=button_dat.employee_df.iloc[0]['job_title'] if not button_dat.employee_df.empty else None, - department=button_dat.employee_df.iloc[0]['department'] if not button_dat.employee_df.empty else None -) +employee = Employee(button_dat) def test_employee_init(): assert employee is not None @@ -34,10 +31,7 @@ def helper_test_reorg_method(): # so will run multiple times in main test initial_state = (employee.job_title, employee.department) - employee.set_current_titles( - job_titles=button_dat.employee_df['job_title'].dropna().values, - departments=button_dat.employee_df['department'].dropna().values - ) + employee.set_current_titles() new_state = (employee.job_title, employee.department) return {"state_change": initial_state != new_state, "job_title": employee.job_title,