Added set_memory_policy tests