| from tools.text_parser import parse_business_input | |
| def test_parse_business_input_with_kwacha_values() -> None: | |
| sample = ( | |
| "I run a small grocery shop in Lusaka. This week I made K4,500 in sales. " | |
| "I spent K2,700 on stock, K500 on rent, K250 on transport, and K300 on other expenses. " | |
| "I also owe K1,000 to my supplier. Can I afford to take a K3,000 loan?" | |
| ) | |
| result = parse_business_input(sample) | |
| assert result["revenue"] == 4500 | |
| assert result["expenses"] == 3750 | |
| assert result["expenses_breakdown"]["stock"] == 2700 | |
| assert result["expenses_breakdown"]["rent"] == 500 | |
| assert result["expenses_breakdown"]["transport"] == 250 | |
| assert result["expenses_breakdown"]["other_expenses"] == 300 | |
| assert result["debt"] == 1000 | |
| assert result["loan_amount"] == 3000 | |
| assert result["business_type"] == "grocery shop" | |
| assert result["location"] == "Lusaka" | |
| def test_parse_business_input_with_plain_numbers() -> None: | |
| sample = "I made 2000 in sales and my total expenses were 1400." | |
| result = parse_business_input(sample) | |
| assert result["revenue"] == 2000 | |
| assert result["expenses"] == 1400 | |