add msg check
diff --git a/tensorflow/python/kernel_tests/cwise_ops_test.py b/tensorflow/python/kernel_tests/cwise_ops_test.py
index 21f5cf2..6420ed4 100644
--- a/tensorflow/python/kernel_tests/cwise_ops_test.py
+++ b/tensorflow/python/kernel_tests/cwise_ops_test.py
@@ -1231,7 +1231,7 @@
   def test_coeffs_raise(self):
     x = np.random.rand(2, 2).astype(np.float32)
     coeffs = {}
-    with self.assertRaises(ValueError):
+    with self.assertRaisesRegexp(ValueError, "Argument coeffs must be list"):
       math_ops.polyval(coeffs, x)