fix: merge

This commit is contained in:
2026-03-17 22:53:12 +08:00
parent cdc2c4ead6
commit ef8ce120bb
3 changed files with 102 additions and 51 deletions

View File

@@ -8,7 +8,11 @@ import pytest
from app.models.transaction import Direction
from app.models.evidence_image import SourceApp
from app.rules.dedup_rules import is_duplicate_pair
from app.rules.transit_rules import is_self_transfer, is_fee_tolerant_transit_pair
from app.rules.transit_rules import (
is_self_transfer,
is_fee_tolerant_transit_pair,
mark_transit_transactions,
)
from app.rules.assessment_rules import classify_transaction
@@ -105,6 +109,36 @@ class TestTransitRules:
)
assert not is_fee_tolerant_transit_pair(out_tx, in_tx)
def test_wechat_recharge_in_then_out_marked_transit(self):
in_tx = _make_tx(
trade_time=datetime(2026, 3, 17, 21, 46, 0, tzinfo=timezone.utc),
amount=50,
direction=Direction.in_,
counterparty_name="零钱充值-来自工商银行(3893)",
counterparty_account="",
self_account_tail_no="3893",
order_no="",
remark="",
confidence=0.95,
is_transit=False,
)
out_tx = _make_tx(
trade_time=datetime(2026, 3, 17, 21, 46, 59, tzinfo=timezone.utc),
amount=50,
direction=Direction.out,
counterparty_name="童年",
counterparty_account="1154****0928",
self_account_tail_no="3893",
order_no="",
remark="充值",
confidence=0.98,
is_transit=False,
)
txs = [in_tx, out_tx]
mark_transit_transactions(txs, known_self_accounts=[])
assert in_tx.is_transit
assert out_tx.is_transit
class TestAssessmentRules:
def test_transit_classified_as_low(self):