File size: 1,169 Bytes
63f0c94 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | 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
|