def test_x_dot_com(self):
self.assertEqual(get_company_names('Zappos.com'),
{'Zappos.com'})
def test_turkish_a_s(self):
# this tests #45
self.assertEqual(get_company_names('Arçelik A.Ş'),
{'Arçelik', 'Arçelik A.Ş'})
def test_comma_llc(self):
# this tests #35
self.assertEqual(get_company_names('Indosole, LLC'),
{'Indosole', 'Indosole, LLC'})
def test_lowercase_a_s(self):
# this tests #30
self.assertEqual(get_company_names('bisgaard sko a/s'),
{'bisgaard sko a/s', 'bisgaard sko'})
def test_pvt_ltd(self):
# this tests #12
self.assertEqual(get_company_names('Servals Pvt Ltd'),
{'Servals', 'Servals Pvt Ltd'})
def test_basic(self):
self.assertEqual(get_company_names('Konica'), {'Konica'})
def test_weird_caching_bug(self):
# would get {'L', 'L Brands'} after calling
# get_company_aliases()
get_company_aliases('L Brands')
self.assertEqual(get_company_names('L Brands'), {'L Brands'})
def test_variant_too_short(self):
self.assertEqual(get_company_names('L Brands'), {'L Brands'})
def test_too_short(self):
self.assertEqual(get_company_names('Y'), set())
def test_empty(self):
self.assertEqual(get_company_names(''), set())
