def test_connection_and_client(self): hook = AzureBatchHook(azure_batch_conn_id=self.test_vm_conn_id) assert isinstance(hook._connection(), Connection) assert isinstance(hook.get_conn(), BatchServiceClient)