pvl/hosts/tests.py
changeset 508 a47849709cbf
parent 507 e3a32f4dff54
child 509 ae53114766fc
--- a/pvl/hosts/tests.py	Thu Feb 26 17:36:55 2015 +0200
+++ b/pvl/hosts/tests.py	Thu Feb 26 17:39:51 2015 +0200
@@ -26,10 +26,14 @@
             self.assertEquals(getattr(host, attr), value)
 
     def assertHostsEqual(self, hosts, expected):
+        hosts = list(hosts)
+
         for host, expect in zip(hosts, expected):
             host_str, attrs = expect
 
             self.assertHostEqual(host, host_str, attrs)
+
+        self.assertEqual(len(hosts), len(expected))
  
     def testApplyHostConfigDict(self):
         host = config.apply_host('foo', 'test', {