Fix and test c10.C10.__repr__().

atac · Apr 15, 2022 · 77da2a7 · 77da2a7
1 parent 3bc9431
commit 77da2a7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/chapter10/c10.py b/chapter10/c10.py
@@ -98,7 +98,7 @@ def __next__(self):
                 self.file.seek(pos + 1)
 
     def __repr__(self):
-        return '<C10: {}>'.format(os.path.basename(self.file.name))
+        return '<C10: {}>'.format(os.path.basename(self.file.io.name))
 
     def __iter__(self):
         return self
diff --git a/tests/unit/test_c10.py b/tests/unit/test_c10.py
@@ -15,3 +15,8 @@ def test_next():
 def test_next_stop():
     with pytest.raises(StopIteration):
         next(c10.C10(Mock(read=Mock(side_effect=EOFError))))
+
+def test_str_repr():
+    c = c10.C10(SAMPLE)
+    repr(c)
+    str(c)