improved extractor unit test

emcf · Aug 29, 2024 · 6739994 · 6739994
1 parent 9f41c9b
commit 6739994
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_extractor.py b/tests/test_extractor.py
@@ -11,7 +11,7 @@
 class TestExtractor(unittest.TestCase):
     def setUp(self):
         self.example_receipt = """# Receipt
-Store: GroceryMart
+Store Name: Grocery Mart
 ## Total
 Subtotal: $13.49 USD
 Tax (8%): $1.08 USD
@@ -45,7 +45,7 @@ def test_extract(self):
             self.assertIn(field, result)
 
         # Check some specific values
-        self.assertEqual(result["store_name"], "GroceryMart")
+        self.assertEqual(result["store_name"], "Grocery Mart")
         self.assertEqual(result["subtotal_usd"], 13.49)
         self.assertEqual(result["tax_usd"], 1.08)
         self.assertEqual(result["total_usd"], 14.57)