remove checks from dict

Author: samuel.oranyeli

janitor/functions/summarize.py

@@ -109,14 +109,7 @@ def summarize(
 
     for num, arg in enumerate(args):
         check(f"Argument {num} in the summarize function", arg, [dict, tuple])
-        if isinstance(arg, dict):
-            for col, func in arg.items():
-                check(
-                    f"func for {col} in argument {num}",
-                    func,
-                    [str, callable],
-                )
-        else:
+        if isinstance(arg, tuple):
             if len(arg) < 2:
                 raise ValueError(
                     f"Argument {num} should have a minimum length of 2, "
@@ -164,6 +157,9 @@ def summarize(
                 val = grp if by_is_true else df
                 if isinstance(func, str):
                     val = val[col]
+                elif is_scalar(func):
+                    aggs[col] = func
+                    break
                 try:
                     outcome = val.agg(func)
                 except (ValueError, AttributeError):

tests/functions/test_summarize.py

@@ -15,17 +15,6 @@ def test_dict_args_error(dataframe):
         dataframe.summarize(1)
 
 
-@pytest.mark.functions
-def test_dict_args_val_error(dataframe):
-    """
-    Raise if arg is a dict,
-    key exist in the columns,
-    but the value is not a string or callable
-    """
-    with pytest.raises(TypeError, match="func for a in argument 0.+"):
-        dataframe.summarize({"a": 1})
-
-
 func = lambda grp: grp.Revenue.sum() / grp.Quantity.sum()  # noqa: E731
 
 
@@ -95,6 +84,8 @@ def test_tuple_func_seq_error(dataframe):
 
 
 args = [
+    {"a": "sum"},
+    {"a": pd.Series([1, 2, 3] * 3).sum()},
     {"a": lambda df: df.a.sum()},
     ("a", "sum"),
     ("a", np.sum),