Implement dose tracking functionality and enhance CSV migration
- Added a new migration script to introduce dose tracking columns in the CSV. - Updated DataManager to handle new dose tracking columns and methods for adding doses. - Enhanced MedTrackerApp to support dose entry and display for each medicine. - Modified UIManager to create a scrollable input frame with dose tracking elements. - Implemented tests for delete functionality, dose tracking, edit functionality, and scrollable input. - Updated existing tests to ensure compatibility with the new CSV format and dose tracking features.
This commit is contained in:
William Valentin
135
scripts/test_edit_window_functionality.py
Normal file
135
scripts/test_edit_window_functionality.py
Normal file
@@ -0,0 +1,135 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Test script to verify edit window functionality (save and delete) after dose tracking
|
||||
implementation.
|
||||
"""
|
||||
|
||||
import logging
|
||||
import os
|
||||
import sys
|
||||
|
||||
# Add the src directory to the path so we can import our modules
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "src"))
|
||||
|
||||
from data_manager import DataManager
|
||||
|
||||
|
||||
def test_edit_window_functionality():
|
||||
"""Test both save and delete functionality with the new CSV format."""
|
||||
print("Testing edit window functionality...")
|
||||
|
||||
# Create a backup of the current CSV
|
||||
import shutil
|
||||
|
||||
try:
|
||||
shutil.copy("thechart_data.csv", "thechart_data_backup.csv")
|
||||
print("✓ Created backup of current CSV")
|
||||
except Exception as e:
|
||||
print(f"✗ Failed to create backup: {e}")
|
||||
return False
|
||||
|
||||
try:
|
||||
# Create a logger for the DataManager
|
||||
logger = logging.getLogger("test_logger")
|
||||
logger.setLevel(logging.DEBUG)
|
||||
|
||||
# Initialize data manager
|
||||
data_manager = DataManager("thechart_data.csv", logger)
|
||||
|
||||
# Load current data
|
||||
df = data_manager.load_data()
|
||||
print(f"✓ Loaded {len(df)} entries from CSV")
|
||||
|
||||
if df.empty:
|
||||
print("✗ No data to test edit functionality")
|
||||
return False
|
||||
|
||||
# Test 1: Test delete functionality
|
||||
print("\n=== Testing Delete Functionality ===")
|
||||
last_entry_date = df.iloc[-1]["date"]
|
||||
print(f"Attempting to delete entry with date: {last_entry_date}")
|
||||
|
||||
success = data_manager.delete_entry(last_entry_date)
|
||||
if success:
|
||||
print("✓ Delete operation successful")
|
||||
df_after_delete = data_manager.load_data()
|
||||
if last_entry_date not in df_after_delete["date"].values:
|
||||
print("✓ Entry successfully removed from CSV")
|
||||
else:
|
||||
print("✗ Entry still exists after delete")
|
||||
return False
|
||||
else:
|
||||
print("✗ Delete operation failed")
|
||||
return False
|
||||
|
||||
# Test 2: Test update functionality
|
||||
print("\n=== Testing Update Functionality ===")
|
||||
if not df_after_delete.empty:
|
||||
# Get first entry to test update
|
||||
first_entry = df_after_delete.iloc[0]
|
||||
test_date = first_entry["date"]
|
||||
original_note = first_entry["note"]
|
||||
print(f"Testing update for date: {test_date}")
|
||||
print(f"Original note: '{original_note}'")
|
||||
|
||||
# Create updated data (simulating what the edit window would do)
|
||||
updated_data = [
|
||||
test_date, # date
|
||||
int(first_entry["depression"]), # depression
|
||||
int(first_entry["anxiety"]), # anxiety
|
||||
int(first_entry["sleep"]), # sleep
|
||||
int(first_entry["appetite"]), # appetite
|
||||
int(first_entry["bupropion"]), # bupropion
|
||||
str(first_entry["bupropion_doses"]), # bupropion_doses
|
||||
int(first_entry["hydroxyzine"]), # hydroxyzine
|
||||
str(first_entry["hydroxyzine_doses"]), # hydroxyzine_doses
|
||||
int(first_entry["gabapentin"]), # gabapentin
|
||||
str(first_entry["gabapentin_doses"]), # gabapentin_doses
|
||||
int(first_entry["propranolol"]), # propranolol
|
||||
str(first_entry["propranolol_doses"]), # propranolol_doses
|
||||
f"{original_note} [UPDATED BY TEST]", # note
|
||||
]
|
||||
|
||||
print(f"Data to update with: {updated_data}")
|
||||
print(f"Length of update data: {len(updated_data)}")
|
||||
|
||||
success = data_manager.update_entry(test_date, updated_data)
|
||||
if success:
|
||||
print("✓ Update operation successful")
|
||||
|
||||
# Verify the update
|
||||
df_after_update = data_manager.load_data()
|
||||
updated_entry = df_after_update[
|
||||
df_after_update["date"] == test_date
|
||||
].iloc[0]
|
||||
if "[UPDATED BY TEST]" in updated_entry["note"]:
|
||||
print("✓ Entry successfully updated in CSV")
|
||||
print(f"New note: '{updated_entry['note']}'")
|
||||
else:
|
||||
print("✗ Entry was not properly updated")
|
||||
return False
|
||||
else:
|
||||
print("✗ Update operation failed")
|
||||
return False
|
||||
|
||||
print("\n✓ All edit window functionality tests passed!")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
print(f"✗ Error during test: {e}")
|
||||
import traceback
|
||||
|
||||
traceback.print_exc()
|
||||
return False
|
||||
|
||||
finally:
|
||||
# Restore the backup
|
||||
try:
|
||||
shutil.move("thechart_data_backup.csv", "thechart_data.csv")
|
||||
print("✓ Restored original CSV from backup")
|
||||
except Exception as e:
|
||||
print(f"✗ Failed to restore backup: {e}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_edit_window_functionality()
|
||||
Reference in New Issue
Block a user