Add tests for 0 to 2 arity memoization