extend coverage

sdpython · Jun 10, 2019 · f24acaf · f24acaf
1 parent 1c6801b
commit f24acaf
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/_unittests/ut_df/test_dataframe_io_helpers.py b/_unittests/ut_df/test_dataframe_io_helpers.py
@@ -214,6 +214,17 @@ def test_read_json_file2(self):
                {'a_a': 3.0, 'a_c': None, 'b_0': 4, 'b_1': 5, 'b_2': 'r'}]
         self.assertEqual(jsjson, exp)
 
+    def test_read_json_item(self):
+        text = TestDataFrameIOHelpers.text_json
+        st = JsonPerRowsStream(StringIO(text))
+        res = []
+        while True:
+            n = st.read()
+            if not n:
+                break
+            res.append(n)
+        self.assertGreater(len(res), 1)
+
 
 if __name__ == "__main__":
     unittest.main()
diff --git a/_unittests/ut_df/test_pandas_groupbynan.py b/_unittests/ut_df/test_pandas_groupbynan.py
@@ -87,6 +87,12 @@ def test_pandas_groupbynan_tuple(self):
             self.assertEqual(d[0]["n"], 1)
             self.assertEqual(d[1]["a"], "NAN")
 
+    def test_pandas_groupbynan_regular(self):
+        df = pandas.DataFrame([dict(a="a", b=1), dict(a="a", b=2)])
+        gr = df.groupby(["a"]).sum()
+        gr2_ = pandas_groupby_nan(df, ["a"]).sum()
+        self.assertEqualDataFrame(gr, gr2_)
+
 
 if __name__ == "__main__":
     unittest.main()