diff -r acdc2cb09f60 -r 3202436d89d0 test/test_nicclient.py --- a/test/test_nicclient.py Thu Jun 02 13:59:46 2016 -0700 +++ b/test/test_nicclient.py Thu Jun 02 14:05:21 2016 -0700 @@ -18,7 +18,6 @@ def test_choose_server(self): domain = 'рнидс.срб' chosen = self.client.choose_server(domain) - self.assertEqual( - chosen, - 'срб'.decode('utf-8').encode('idna') + '.whois-servers.net' - ) + suffix = domain.split('.')[-1].encode('idna').decode('utf-8') + correct = '{}.whois-servers.net'.format(suffix) + self.assertEqual(chosen, correct)