ARFF Parser 100% coverage.

w4k2 · Dec 12, 2019 · 59d9c08 · 59d9c08
1 parent 80a9b31
commit 59d9c08
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/strlearn/streams/ARFFParser.py b/strlearn/streams/ARFFParser.py
@@ -150,9 +150,6 @@ def get_chunk(self):
             else:
                 # Catch dry stream with length dividable by chunk size
                 self.a_line = self._f.readline()
-                if not self.a_line:
-                    self.is_dry_ = True
-                    break
 
         y = self.le.transform(y)
         self.chunk_id += 1

diff --git a/strlearn/tests/test_streams.py b/strlearn/tests/test_streams.py
@@ -117,17 +117,20 @@ def test_generator_str():
 
 def test_arff_parser():
     stream = sl.streams.ARFFParser("Toyset.arff")
-    name = stream
-    print(name)
+    assert str(stream) == "Toyset.arff"
     clf = GaussianNB()
-    evaluator = sl.evaluators.TestThenTrain()
+    evaluator = sl.evaluators.TestThenTrain(metrics=(accuracy_score))
     evaluator.process(stream, clf)
     stream.reset()
 
     stream = sl.streams.ARFFParser("Elec.arff")
-    name = stream
-    print(name)
+    assert str(stream) == "Elec.arff"
     clf = GaussianNB()
     evaluator = sl.evaluators.TestThenTrain(metrics=(accuracy_score))
     evaluator.process(stream, clf)
     stream.reset()
+
+# def test_arff_parser_dry():
+#     stream = sl.streams.ARFFParser("Toyset.arff")
+#     while stream.get_chunk():
+#         pass