update chr test

Ailing Zhang · Ailing Zhang · commit 18bd545d2adc · 2019-06-14T11:09:56.000-07:00
diff --git a/test/repro.py b/test/repro.py
@@ -0,0 +1,9 @@
+import torch
+
+@torch.jit.script
+def func(x):
+    # type: (int) -> int
+    print(chr(x))
+    return x
+print(func(255))
+print(chr(255))
diff --git a/test/test_jit.py b/test/test_jit.py
@@ -11971,16 +11971,13 @@ def index_str_to_tensor(s):
         s = u'\u00a3'.encode('utf8')[:1]
         self.checkScript(index_str_to_tensor, (s,))
 
-    @unittest.skipIf(not PY2, "We don't support returning wstring")
     def test_chr(self):
         def fn(x):
             # type: (int) -> str
             return chr(x)
 
         self.checkScript(fn, (1,))
-        self.checkScript(fn, (255,))
-
-        self.checkScriptRaisesRegex(fn, (256,), Exception, 'not in range(256)')
+        self.checkScript(fn, (97,))
 
     def test_round(self):
         def round_float(x):
diff --git a/torch/csrc/jit/register_prim_ops.cpp b/torch/csrc/jit/register_prim_ops.cpp
@@ -2165,12 +2165,14 @@ RegisterOperators reg2({
         "aten::chr(int i) -> str",
         [](Stack& stack) {
           auto i = pop(stack).toInt();
+          std::stringstream ss;
           TORCH_CHECK(
               i >= 0 && i < 1114111,
-              "chr() arg not in range(256), found ",
+              "chr() arg not in range(0x110000), found ",
               i);
           char c = i;
-          push(stack, std::string(c, 1));
+          ss << c;
+          push(stack, ss.str());
           return 0;
         }),
 #define CREATE_COPY_OP(other_type, c_type)                                 \
