Adding memoization unit tests