TST: Add tests for Python2, Python3 *.npy compatibility.

numpy · Oct 12, 2014 · 8b1f90a · 8b1f90a
1 parent 8f068b7
commit 8b1f90a
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 0 deletions.
diff --git a/.gitattributes b/.gitattributes
@@ -1,2 +1,5 @@
 * text=auto
 tools/win32build/nsis_scripts/*.nsi.in eol=crlf
+
+# Numerical data files
+numpy/lib/tests/data/*.npy binary
diff --git a/numpy/lib/tests/data/python3.npy b/numpy/lib/tests/data/python3.npy
diff --git a/numpy/lib/tests/data/win64python2.npy b/numpy/lib/tests/data/win64python2.npy
diff --git a/numpy/lib/tests/test_format.py b/numpy/lib/tests/test_format.py
@@ -524,6 +524,16 @@ def test_compressed_roundtrip():
     assert_array_equal(arr, arr1)
 
 
+def test_python2_python3_interoperability():
+    if sys.version_info[0] >= 3:
+        fname = 'win64python2.npy'
+    else:
+        fname = 'python3.npy'
+    path = os.path.join(os.path.dirname(__file__), 'data', fname)
+    data = np.load(path)
+    assert_array_equal(data, np.ones(2))
+
+
 def test_version_2_0():
     f = BytesIO()
     # requires more than 2 byte for header