`self.get_total_num_groups()` and be replaced with `current_group_count`. This call is costly and will slow the test significantly.