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

@@ -179,3 +179,18 @@ def test():
         callable=test,
         expect="""mm(): cannot matrix-multiply tensors f32[2,5] and f32[8,2]. Expected the size of dimension 1 of the first input tensor (5) to be equal the size of dimension 0 of the second input tensor (8)."""
     )
+
+  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)."""
+    )