def test_custom_client(self): class MyClient(Client): pass client = PooledClient(('host', 11211)) client.client_class = MyClient assert isinstance(client.client_pool.get(), MyClient)