TST: Change expected output for Python 3.

migueldvb · Sep 26, 2012 · 0f5bbfa · 0f5bbfa
1 parent 8e0a542
commit 0f5bbfa
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/numpy/core/tests/test_arrayprint.py b/numpy/core/tests/test_arrayprint.py
@@ -1,3 +1,5 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
 import sys
 import numpy as np
 from numpy.testing import *
@@ -148,8 +150,13 @@ def test_formatter_reset(self):
         assert_equal(repr(x), "array([ 0.,  1.,  2.])")
 
 def test_unicode_object_array():
+    import sys
+    if sys.version_info[0] >= 3:
+        expected = "array(['é'], dtype=object)"
+    else:
+        expected = "array([u'\\xe9'], dtype=object)"
     x = np.array([u'\xe9'], dtype=object)
-    assert_equal(repr(x), "array([u'\\xe9'], dtype=object)")
+    assert_equal(repr(x), expected)