code: def test_rescale_read_counts(
      self, n_ref, n_total, max_allowed_reads, expected
  ):
      actual = variant_caller._rescale_read_counts_if_necessary(
          n_ref, n_total, max_allowed_reads
      )
      self.assertEqual(actual, expected)
  
imports: from unittest import mock
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  import numpy.testing
  from third_party.nucleus.util import variant_utils
  from third_party.nucleus.util import variantcall_utils
  from deepvariant import testdata
  from deepvariant import variant_caller
  from deepvariant.protos import deepvariant_pb2
label_desc: Use image node_id 3 for calling the VariantCallerTests obj's underlying member method code with example usage: obj.test_rescale_read_counts(n_ref, n_total, max_allowed_reads, expected) without return types