def test_append(self): metric_a = ToyConversationMetric(self.conv, metric_name='a') metrics = ConversationMetricSet(self.conv, [metric_a]) self.assertEqual(metrics.values(), [metric_a]) metric_b = ToyConversationMetric(self.conv, metric_name='b') metrics.append(metric_b) self.assertEqual(metrics.values(), [metric_a, metric_b]) self.assertEqual(metrics['b'], metric_b)
def test_extend(self): metric_a = ToyConversationMetric(self.conv, metric_name='a') metrics = ConversationMetricSet(self.conv, [metric_a]) self.assertEqual(metrics.values(), [metric_a]) metric_b = ToyConversationMetric(self.conv, metric_name='b') metric_c = ToyConversationMetric(self.conv, metric_name='c') metrics.extend([metric_b, metric_c]) self.assertEqual(metrics.values(), [metric_a, metric_b, metric_c]) self.assertEqual(metrics['b'], metric_b) self.assertEqual(metrics['c'], metric_c)
def test_iteration(self): metric_a = ToyConversationMetric(self.conv, metric_name='a') metric_b = ToyConversationMetric(self.conv, metric_name='b') metrics = ConversationMetricSet(self.conv, [metric_a, metric_b]) self.assertEqual(list(iter(metrics.values())), [metric_a, metric_b])