diff options
Diffstat (limited to 'ttystatus/bytespeed_tests.py')
-rw-r--r-- | ttystatus/bytespeed_tests.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/ttystatus/bytespeed_tests.py b/ttystatus/bytespeed_tests.py index 369c2f6..b56afc3 100644 --- a/ttystatus/bytespeed_tests.py +++ b/ttystatus/bytespeed_tests.py @@ -25,30 +25,30 @@ class ByteSpeedTests(unittest.TestCase): self.w = ttystatus.ByteSpeed('foo') def test_formats_zero_speed_without_update(self): - self.assertEqual(str(self.w), '0 B/s') + self.assertEqual(self.w.render(), '0 B/s') def test_formats_zero_bytes_correctly(self): self.w.update({ 'foo': 0 }) - self.assertEqual(str(self.w), '0 B/s') + self.assertEqual(self.w.render(), '0 B/s') def test_formats_one_byte_per_second_correctly(self): self.w.now = lambda: 1 self.w.update({ 'foo': 0 }) self.w.now = lambda: 2 self.w.update({ 'foo': 1 }) - self.assertEqual(str(self.w), '1 B/s') + self.assertEqual(self.w.render(), '1 B/s') def test_formats_ten_bytes_per_second_correctly(self): self.w.now = lambda: 1 self.w.update({ 'foo': 0 }) self.w.now = lambda: 11 self.w.update({ 'foo': 100 }) - self.assertEqual(str(self.w), '10 B/s') + self.assertEqual(self.w.render(), '10 B/s') def test_formats_ten_tibs_per_second_correctly(self): self.w.now = lambda: 1 self.w.update({ 'foo': 0 }) self.w.now = lambda: 2 self.w.update({ 'foo': 10 * 1024**4 }) - self.assertEqual(str(self.w), '10.00 TiB/s') + self.assertEqual(self.w.render(), '10.00 TiB/s') |