diff --git a/tests/test_config.py b/tests/test_config.py index c1101b88..c72ae605 100644 --- a/tests/test_config.py +++ b/tests/test_config.py @@ -33,10 +33,14 @@ def test_init(self): expected_experiment_key_map['group_exp_2']['groupId'] = '19228' expected_experiment_key_map['group_exp_2']['groupPolicy'] = 'random' expected_experiment_id_map = { - 'test_experiment': self.config_dict['experiments'][0], - 'group_exp_1': self.config_dict['groups'][0]['experiments'][0], - 'group_exp_2': self.config_dict['groups'][0]['experiments'][1] + '111127': self.config_dict['experiments'][0], + '32222': self.config_dict['groups'][0]['experiments'][0], + '32223': self.config_dict['groups'][0]['experiments'][1] } + expected_experiment_id_map['32222']['groupId'] = '19228' + expected_experiment_id_map['32222']['groupPolicy'] = 'random' + expected_experiment_id_map['32223']['groupId'] = '19228' + expected_experiment_id_map['32223']['groupPolicy'] = 'random' expected_event_key_map = { 'test_event': self.config_dict['events'][0], 'Total Revenue': self.config_dict['events'][1] @@ -115,9 +119,10 @@ def test_init(self): } } } + self.assertEqual(expected_group_id_map, self.project_config.group_id_map) self.assertEqual(expected_experiment_key_map, self.project_config.experiment_key_map) - self.assertEqual(expected_experiment_id_map, self.project_config.experiment_key_map) + self.assertEqual(expected_experiment_id_map, self.project_config.experiment_id_map) self.assertEqual(expected_event_key_map, self.project_config.event_key_map) self.assertEqual(expected_attribute_key_map, self.project_config.attribute_key_map) self.assertEqual(expected_audience_id_map, self.project_config.audience_id_map)