def test_set_memcache_timeout_policy(testbed): def policy(key): return 1 snippets.set_memcache_timeout_policy(policy) assert ndb.get_context().get_memcache_timeout_policy() == policy