Move test.

Author: ysiraichi

test/test_operations.py

@@ -2367,19 +2367,6 @@ def test_isneginf_no_fallback(self):
     t = t.to(torch.float16)
     self._test_no_fallback(torch.isneginf, (t,))
 
-  def test_uniform__raises_error_on_invalid_range(self):
-    device = torch_xla.device()
-    a = torch.empty(5, 5, device=device)
-    from_ = 5.
-    to_ = 2.
-
-    try:
-      a.uniform_(from_, to_)
-    except RuntimeError as e:
-      expected_error = (
-          "uniform_(): expected `from` (5) to be smaller or equal `to` (2).")
-      self.assertEqual(str(e), expected_error)
-
 
 class MNISTComparator(nn.Module):
 

test/test_ops_error_message.py

@@ -222,6 +222,21 @@ def test():
         expect="""roll(): expected `dims` [0] (size=1) to match the size of `shifts` [2, 2] (size=2)."""
     )
 
+  def test_uniform__raises_error_on_invalid_range(self):
+    device = torch_xla.device()
+    a = torch.empty(5, 5, device=device)
+    from_ = 5.
+    to_ = 2.
+
+    def test():
+      return a.uniform_(from_, to_)
+
+    self.assertExpectedRaisesInline(
+        exc_type=RuntimeError,
+        callable=test,
+        expect="""uniform_(): expected `from` (5) to be smaller or equal `to` (2)."""
+    )
+
 
 if __name__ == "__main__":
   unittest.main()