def test_rules_in_groups(self):
segment = Segment(name="Bar")
rule_1 = Rule(group_id=1, field='foo', comparator="MATCH", value='bar')
rule_2 = Rule(group_id=2, field='foo', comparator="MATCH", value='bar')
rule_3 = Rule(group_id=2, field='foo', comparator="MATCH", value='bar')
rule_4 = Rule(group_id=3, field='foo', comparator="MATCH", value='bar')
rule_5 = Rule(group_id=3, field='foo', comparator="MATCH", value='bar')
rule_6 = Rule(group_id=3, field='foo', comparator="MATCH", value='bar')
segment.rules = [rule_1, rule_2, rule_3, rule_4, rule_5, rule_6]
groups = segment.rules_in_groups()
self.assertEqual(len(groups), 3)
self.assertEqual(groups[1], [rule_1])
self.assertEqual(groups[2], [rule_2, rule_3])
self.assertEqual(groups[3], [rule_4, rule_5, rule_6])
