'Tests the use of multiple models with the context manager
@since 3.7
@jira_ticket PYTHON-613
@expected_result all models are properly updated with the context
@test_category query'

def test_context_multiple_models(self):

    with ContextQuery(TestModel, TestModel, keyspace='ks4') as (tm1, tm2):
        self.assertNotEqual(tm1, tm2)
        self.assertEqual(tm1.__keyspace__, 'ks4')
        self.assertEqual(tm2.__keyspace__, 'ks4')
