include missing close call in test_NamedTupleWriter() assertion

xflr6 · Jul 22, 2023 · d212635 · d212635
1 parent ffad534
commit d212635
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_extras.py b/tests/test_extras.py
@@ -91,4 +91,5 @@ def test_NamedTupleWriter(mocker, rows, lines):
         writer = NamedTupleWriter(f)
         assert writer.dialect.delimiter == ','
         writer.writerows(rows)
-    assert f.method_calls == [mocker.call.write(l) for l in lines]
+    expected = [mocker.call.write(l) for l in lines] + [mocker.call.close()]
+    assert f.method_calls == expected