update test for the `memory_on_functions` factory