fix: merge
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user