Merge pull request #1139 from roboflow/tests/cover-collect-func-params-with-tests

Cover collect_func_params with test

Author: grzegorz-roboflow

tests/inference/unit_tests/usage_tracking/test_collect_func_params.py

@@ -0,0 +1,121 @@
+from inference.usage_tracking.utils import collect_func_params
+
+
+def test_collect_func_params():
+    # given
+    def func(arg1, arg2, kwarg3, kwarg4):
+        pass
+
+    # when
+    func_params = collect_func_params(
+        func=func,
+        args=["val1", "val2"],
+        kwargs={"kwarg3": "val3", "kwarg4": "val4"},
+    )
+
+    # then
+    assert func_params == {
+        "arg1": "val1",
+        "arg2": "val2",
+        "kwarg3": "val3",
+        "kwarg4": "val4",
+    }
+
+
+def test_collect_func_params_with_default_args():
+    # given
+    def func(
+        arg1,
+        arg2="default val2",
+        arg3="default val3",
+        kwarg4="default val4",
+        kwarg5="default val5",
+    ):
+        pass
+
+    # when
+    func_params = collect_func_params(
+        func=func,
+        args=["val1", "val2"],
+        kwargs={"kwarg4": "val4"},
+    )
+
+    # then
+    assert func_params == {
+        "arg1": "val1",
+        "arg2": "val2",
+        "arg3": "default val3",
+        "kwarg4": "val4",
+        "kwarg5": "default val5",
+    }
+
+
+def test_collect_func_params_with_args_and_catchall_kwargs():
+    # given
+    def func(
+        arg1,
+        arg2="default val2",
+        arg3="default val3",
+        kwarg4="default val4",
+        kwarg5="default val5",
+        **kwargs
+    ):
+        pass
+
+    # when
+    func_params = collect_func_params(
+        func=func,
+        args=["val1", "val2"],
+        kwargs={"kwarg4": "val4", "kwarg6": "val6"},
+    )
+
+    # then
+    assert func_params == {
+        "arg1": "val1",
+        "arg2": "val2",
+        "arg3": "default val3",
+        "kwarg4": "val4",
+        "kwarg5": "default val5",
+        "kwarg6": "val6",
+        "kwargs": {"kwarg4": "val4", "kwarg6": "val6"},
+    }
+
+
+def test_collect_func_params_with_only_kwargs():
+    # given
+    def func(**kwargs):
+        pass
+
+    # when
+    func_params = collect_func_params(
+        func=func,
+        args=[],
+        kwargs={"kwarg4": "val4", "kwarg6": "val6"},
+    )
+
+    # then
+    assert func_params == {
+        "kwargs": {"kwarg4": "val4", "kwarg6": "val6"},
+        "kwarg4": "val4",
+        "kwarg6": "val6",
+    }
+
+
+def test_collect_func_params_with_only_kwargs_and_errorneous_args_passed():
+    # given
+    def func(**kwargs):
+        pass
+
+    # when
+    func_params = collect_func_params(
+        func=func,
+        args=["val1", "val2"],
+        kwargs={"kwarg4": "val4", "kwarg6": "val6"},
+    )
+
+    # then
+    assert func_params == {
+        "kwargs": {"kwarg4": "val4", "kwarg6": "val6"},
+        "kwarg4": "val4",
+        "kwarg6": "val6",
+    }, "Unexpected args should be ignored rather than causing errors"