def test_non_key(self): with self.assertRaises(ValueError): switch_average('E')
def test_key_C(self): self.assertEqual(switch_average('C'), 98)
def test_key_D(self): self.assertEqual(switch_average('D'), 90)
def test_key_B(self): self.assertEqual(switch_average('B'), 95)