*** test/record/test_legacy_records.py	Mon Sep 27 01:35:48 2021
--- test/record/test_legacy_records.py	Mon Sep 27 01:38:01 2021
***************
*** 1,6 ****
  from __future__ import unicode_literals
  import pytest
! from mock import patch
  from kafka.record.legacy_records import (
      LegacyRecordBatch, LegacyRecordBatchBuilder
  )
--- 1,10 ----
  from __future__ import unicode_literals
  import pytest
! try:
!     from unittest.mock import patch
! except ImportError:
!     from mock import patch
! 
  from kafka.record.legacy_records import (
      LegacyRecordBatch, LegacyRecordBatchBuilder
  )