add close after using stringio

usc-isi-i2 · Nov 21, 2018 · 57ef371 · 57ef371
1 parent e83474c
commit 57ef371
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/rltk/tests/test_io_reader.py b/rltk/tests/test_io_reader.py
@@ -22,6 +22,7 @@ def test_dataframe_reader():
 
 def test_csv_reader():
     f = io.StringIO()
+
     writer = csv.DictWriter(f, fieldnames=['1', '2'])
     writer.writeheader()
     for a in arr:
@@ -35,8 +36,11 @@ def test_csv_reader():
 
 def test_jsonlines_reader():
     f = io.StringIO()
+
     for a in arr:
         f.write(json.dumps(a) + '\n')
 
     for idx, obj in enumerate(JsonLinesReader(f)):
         assert obj == arr[idx]
+
+    f.close()