|
|
""" |
|
|
Tests for FinEE Regex Engine (Tier 1). |
|
|
""" |
|
|
|
|
|
import pytest |
|
|
from finee.regex_engine import get_regex_engine, TransactionType |
|
|
|
|
|
@pytest.fixture |
|
|
def engine(): |
|
|
return get_regex_engine() |
|
|
|
|
|
def test_amount_extraction(engine): |
|
|
"""Test amount extraction patterns.""" |
|
|
cases = [ |
|
|
("Rs.500.00 debited", 500.0), |
|
|
("INR 2500 debited", 2500.0), |
|
|
("Rs 1,234.50 spent", 1234.5), |
|
|
("Amt: 50.00", 50.0), |
|
|
("Amount: 100", 100.0), |
|
|
("debited 200.00 from", 200.0), |
|
|
] |
|
|
for text, expected in cases: |
|
|
result = engine.extract(text) |
|
|
assert result.amount == expected, f"Failed on: {text}" |
|
|
|
|
|
def test_date_extraction(engine): |
|
|
"""Test date extraction patterns.""" |
|
|
cases = [ |
|
|
("on 01-01-2025", "01-01-2025"), |
|
|
("on 01/01/25", "01/01/25"), |
|
|
("on 28 Dec 2025", "28 Dec 2025"), |
|
|
("Date: 15-08-2024", "15-08-2024"), |
|
|
] |
|
|
for text, expected in cases: |
|
|
result = engine.extract(text) |
|
|
assert result.date == expected, f"Failed on: {text}" |
|
|
|
|
|
def test_type_extraction(engine): |
|
|
"""Test transaction type extraction.""" |
|
|
debit_cases = [ |
|
|
"debited from", "starting debit", "spent on", "paid to", "sent to" |
|
|
] |
|
|
credit_cases = [ |
|
|
"credited to", "received from", "refund of", "cashback received", "reversed" |
|
|
] |
|
|
|
|
|
for text in debit_cases: |
|
|
assert engine.extract(text).type == TransactionType.DEBIT, f"Failed debit: {text}" |
|
|
|
|
|
for text in credit_cases: |
|
|
assert engine.extract(text).type == TransactionType.CREDIT, f"Failed credit: {text}" |
|
|
|
|
|
def test_account_extraction(engine): |
|
|
"""Test account number extraction.""" |
|
|
cases = [ |
|
|
("A/c 1234", "1234"), |
|
|
("Account XXXXX1234", "1234"), |
|
|
("ending with 5678", "5678"), |
|
|
("A/c no. 4321", "4321"), |
|
|
] |
|
|
for text, expected in cases: |
|
|
result = engine.extract(text) |
|
|
assert result.account == expected, f"Failed on: {text}" |
|
|
|
|
|
def test_reference_extraction(engine): |
|
|
"""Test reference/UTR extraction.""" |
|
|
cases = [ |
|
|
("Ref: 123456789012", "123456789012"), |
|
|
("UTR: 098765432109", "098765432109"), |
|
|
("UPI Ref 112233445566", "112233445566"), |
|
|
] |
|
|
for text, expected in cases: |
|
|
result = engine.extract(text) |
|
|
assert result.reference == expected, f"Failed on: {text}" |
|
|
|
|
|
def test_vpa_extraction(engine): |
|
|
"""Test UPI VPA extraction.""" |
|
|
cases = [ |
|
|
("to swiggy@ybl", "swiggy@ybl"), |
|
|
("VPA: merchant@okaxis", "merchant@okaxis"), |
|
|
("from user@sbi", "user@sbi"), |
|
|
] |
|
|
for text, expected in cases: |
|
|
result = engine.extract(text) |
|
|
assert result.vpa == expected, f"Failed on: {text}" |
|
|
|
|
|
def test_complex_message(engine): |
|
|
"""Test a full banking message.""" |
|
|
text = "Rs.2500.00 debited from HDFC A/c 3545 to VPA swiggy@ybl on 28-12-25. Ref: 534567891234" |
|
|
result = engine.extract(text) |
|
|
|
|
|
assert result.amount == 2500.0 |
|
|
assert result.type == TransactionType.DEBIT |
|
|
assert result.account == "3545" |
|
|
assert result.vpa == "swiggy@ybl" |
|
|
assert result.date == "28-12-25" |
|
|
assert result.reference == "534567891234" |
|
|
assert result.bank == "HDFC" |
|
|
|
