def test_scrape_account_no_match(): """Does the account number get extracted for a bad filename?""" path = os.path.join("/", "arbitrary", "dir", Bbt.FILE_PREFIX + "not 4 digits") assert Bbt.parse_account_number(path) is None
def test_scrape_account(): """Does the account number get extracted from the filename?""" path = os.path.join("/", "arbitrary", "dir", expected_filename()) account = Bbt.parse_account_number(path) assert FAKE_BBT_ACCOUNT == int(account)