*** test/record/test_default_records.py		Mon Sep 27 03:33:57 2021
--- test/record/test_default_records.py    	Mon Sep 27 03:32:47 2021
***************
*** 1,7 ****
  # -*- coding: utf-8 -*-
  from __future__ import unicode_literals
  import pytest
! from mock import patch
  import kafka.codec
  from kafka.record.default_records import (
      DefaultRecordBatch, DefaultRecordBatchBuilder
--- 1,10 ----
  # -*- coding: utf-8 -*-
  from __future__ import unicode_literals
  import pytest
! try:
!     from unittest.mock import patch
! except ImportError:
!     from mock import patch
  import kafka.codec
  from kafka.record.default_records import (
      DefaultRecordBatch, DefaultRecordBatchBuilder