def METHOD_NAME(self, batch_size):
    stats = self.client_.get_inference_statistics(self.model_name_, "1")
    self.assertEqual(len(stats.model_stats), 1, "expect 1 model stats")
    self.assertEqual(
        stats.model_stats[0].name,
        self.model_name_,
        "expect model stats for model {}".format(self.model_name_),
    )
    self.assertEqual(
        stats.model_stats[0].version,
        "1",
        "expect model stats for model {} version 1".format(self.model_name_),
    )
    batch_stats = stats.model_stats[0].batch_stats
    batch_input_duration = 0
    for batch_stat in batch_stats:
        if batch_stat.batch_size == batch_size:
            batch_input_duration = batch_stat.compute_input.ns
    return batch_input_duration