File size: 3,178 Bytes
dcc24f8 |
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 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
"""
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"
|