fixes to Edit functionality

Author: hoffm386

Lesson2-Exercise2-Delete/solution/src/tracker.py

@@ -29,7 +29,7 @@ def edit_expense(self, expense_id, new_amount=None, new_category=None):
         expense = self.get_expense(expense_id)
         if expense:
             if new_amount is not None:
-                expense["amount"] = amount
+                expense["amount"] = new_amount
             if new_category is not None:
                 expense["category"] = new_category
 

Lesson2-Exercise2-Delete/solution/tests/test_tracker.py

@@ -55,7 +55,7 @@ def test_edit_expense(tracker):
     edited_expense = tracker.get_expense(expense_id)
     # Assert
     assert tracker.total == 12.50
-    assert edit_expense["category"] == "dining"
+    assert edited_expense["category"] == "dining"
 
 def test_delete_expense(tracker):    
     # Arrange: Add an expense to delete

Lesson2-Exercise2-Delete/starter/src/tracker.py

@@ -29,7 +29,7 @@ def edit_expense(self, expense_id, new_amount=None, new_category=None):
         expense = self.get_expense(expense_id)
         if expense:
             if new_amount is not None:
-                expense["amount"] = amount
+                expense["amount"] = new_amount
             if new_category is not None:
                 expense["category"] = new_category
 

Lesson2-Exercise2-Delete/starter/tests/test_tracker.py

@@ -47,7 +47,7 @@ def test_edit_expense(tracker):
     edited_expense = tracker.get_expense(expense_id)
     # Assert
     assert tracker.total == 12.50
-    assert edit_expense["category"] == "dining"
+    assert edited_expense["category"] == "dining"
 
 def test_list_expenses_by_category_returns_only_requested_category(tracker):
     # Arrange