pvl.hosts.test: fix assertHostsEqual to also compare len's
authorTero Marttila <tero.marttila@aalto.fi>
Thu, 26 Feb 2015 17:39:51 +0200
changeset 508 a47849709cbf
parent 507 e3a32f4dff54
child 509 ae53114766fc
pvl.hosts.test: fix assertHostsEqual to also compare len's
pvl/hosts/tests.py
--- 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', {