test note and normalize asserts

aholmberg · aholmberg · commit 40e7e63a980b · 2016-05-26T12:28:28.000-05:00
diff --git a/tests/unit/test_metadata.py b/tests/unit/test_metadata.py
@@ -480,13 +480,16 @@ def test_user_type(self):
 
 class HostsTests(unittest.TestCase):
     def test_iterate_all_hosts_and_modify(self):
+        """
+        PYTHON-572
+        """
         metadata = Metadata()
         metadata.add_or_return_host(Host('dc1.1', SimpleConvictionPolicy))
         metadata.add_or_return_host(Host('dc1.2', SimpleConvictionPolicy))
 
-        assert len(metadata.all_hosts()) == 2
+        self.assertEqual(len(metadata.all_hosts()), 2)
 
-        for host in metadata.all_hosts():
+        for host in metadata.all_hosts():  # this would previously raise in Py3
             metadata.remove_host(host)
 
-        assert len(metadata.all_hosts()) == 0
+        self.assertEqual(len(metadata.all_hosts()), 0)
