code: def check_group_assignment(
      self, count_fn, groups_a, groups_b, correct_idxs, correct_counts
  ):
      assign_idxs, counts = calc_group_assignment(
          groups_a, groups_b, count_fn
      )
      self.assertEqual(assign_idxs, correct_idxs)
      self.assertEqual(counts, correct_counts)
  
imports: import unittest
  from copy import deepcopy
  from finetune.util.group_metrics import get_count_fn, calc_group_assignment, create_joint_groups, get_metrics, group_metrics, joint_metrics
label_desc: Use image node_id 4 for calling the TestGroupMetrics obj's underlying member method code with example usage: obj.check_group_assignment(count_fn, groups_a, groups_b, correct_idxs, correct_counts) without return types