cleanup of deprecated test methods

PiperOrigin-RevId: 708282906
google · Dec 23, 2024 · 424f364 · 424f364
1 parent 723f839
commit 424f364
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/edward2/tensorflow/random_variable_test.py b/edward2/tensorflow/random_variable_test.py
@@ -103,7 +103,7 @@ def testStr(self):
     x = ed.RandomVariable(tfp.distributions.Normal(0.0, 1.0), value=1.234)
     pattern = "RandomVariable(\"1.234\", shape=(), dtype=float32"
     regexp = re.escape(pattern)
-    self.assertRegexpMatches(str(x), regexp)
+    self.assertRegex(str(x), regexp)
 
   def testRepr(self):
     x = ed.RandomVariable(tfp.distributions.Normal(0.0, 1.0), value=1.234)