tests: simplify tests a bit
tests: add test for function argument