def test_check_company_ok(self):
company = ' Resultados Digitais S./A. '
company = SFNL.check_company(company)
assert (company == 'Resultados Digitais S./A.')
company = 'Oi! '
company = SFNL.check_company(company)
assert (company == 'Oi!')
company = 'Viva :)'
company = SFNL.check_company(company)
assert (company == 'Viva :)')
def test_check_company_not_ok(self):
company = 12345
try:
company = SFNL.check_company(company)
raise AssertionError()
except Exception as invalid:
print(invalid)
company = ' 12345 '
try:
company = SFNL.check_company(company)
raise AssertionError()
except Exception as invalid:
print(invalid)
company = ''
try:
company = SFNL.check_company(company)
raise AssertionError()
except Exception as invalid:
print(invalid)