def expect(payload, events, expected_attn):
     self.assertEqual(classifier.calculate_attention(events, payload),
                      expected_attn)
 def expect(payload, events, expected_attn):
     self.assertEqual(classifier.calculate_attention(events, payload),
                      expected_attn)