tests for dicts and result_type='infer'

Author: jreback

pandas/tests/frame/test_apply.py

@@ -509,14 +509,16 @@ def test_infer_row_shape(self):
 
     def test_with_dictlike_columns(self):
         # gh 17602
-
         df = DataFrame([[1, 2], [1, 2]], columns=['a', 'b'])
-        result = df.apply(lambda x: {'s': x['a'] + x['b']}, 1)
+        result = df.apply(lambda x: {'s': x['a'] + x['b']},
+                          axis=1)
         expected = Series([{'s': 3} for t in df.itertuples()])
         assert_series_equal(result, expected)
 
         df['tm'] = [pd.Timestamp('2017-05-01 00:00:00'),
                     pd.Timestamp('2017-05-02 00:00:00')]
+        result = df.apply(lambda x: {'s': x['a'] + x['b']},
+                          axis=1)
         assert_series_equal(result, expected)
 
         # compose a series
@@ -535,6 +537,20 @@ def test_with_dictlike_columns(self):
         expected = Series([{}, {}, {}])
         assert_series_equal(result, expected)
 
+    def test_with_dictlike_columns_with_infer(self):
+        # gh 17602
+        df = DataFrame([[1, 2], [1, 2]], columns=['a', 'b'])
+        result = df.apply(lambda x: {'s': x['a'] + x['b']},
+                          axis=1, result_type='infer')
+        expected = DataFrame({'s': [3, 3]})
+        assert_frame_equal(result, expected)
+
+        df['tm'] = [pd.Timestamp('2017-05-01 00:00:00'),
+                    pd.Timestamp('2017-05-02 00:00:00')]
+        result = df.apply(lambda x: {'s': x['a'] + x['b']},
+                          axis=1, result_type='infer')
+        assert_frame_equal(result, expected)
+
     def test_with_listlike_columns(self):
         # gh-17348
         df = DataFrame({'a': Series(np.random.randn(4)),