Simplification of memoize expected case
also specs the case mentioned by @jdalton in comments (great catch!)