Update assert style in test.

waveaccounting · Feb 26, 2015 · c45a6b7 · c45a6b7
1 parent 6974f40
commit c45a6b7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests.py b/tests.py
@@ -146,13 +146,15 @@ def test_persistance(self):
         storage_a = InMemoryStorage()
         storage_b = InMemoryStorage()
 
-        assert storage_a.filesystem is storage_b.filesystem
+        # Can't use self.assertIs because it isn't available in Python 2.6
+        self.assertTrue(storage_a.filesystem is storage_b.filesystem)
 
     def test_no_persistance_without_setting(self):
         storage_a = InMemoryStorage()
         storage_b = InMemoryStorage()
 
-        assert storage_a.filesystem is not storage_b.filesystem
+        # Can't use self.assertIs because it isn't available in Python 2.6
+        self.assertTrue(storage_a.filesystem is not storage_b.filesystem)
 
 if __name__ == '__main__':
     unittest.main()