Update scope of I/O tests

Author: antonwolfy

dpnp/tests/third_party/cupy/io_tests/test_formatting.py

@@ -0,0 +1,47 @@
+import unittest
+
+import numpy
+import pytest
+
+import dpnp as cupy
+from dpnp.tests.third_party.cupy import testing
+
+pytest.skip("text formatting is not supported yet", allow_module_level=True)
+
+
+class TestFormatting(unittest.TestCase):
+
+    def test_array_repr(self):
+        a = testing.shaped_arange((2, 3, 4), cupy)
+        b = testing.shaped_arange((2, 3, 4), numpy)
+        assert cupy.array_repr(a) == numpy.array_repr(b)
+
+    def test_array_str(self):
+        a = testing.shaped_arange((2, 3, 4), cupy)
+        b = testing.shaped_arange((2, 3, 4), numpy)
+        assert cupy.array_str(a) == numpy.array_str(b)
+
+    def test_array2string(self):
+        a = testing.shaped_arange((2, 3, 4), cupy)
+        b = testing.shaped_arange((2, 3, 4), numpy)
+        assert cupy.array2string(a) == numpy.array2string(b)
+
+    def test_format_float_positional_python_scalar(self):
+        x = 1.0
+        assert cupy.format_float_positional(x) == numpy.format_float_positional(
+            x
+        )
+
+    def test_format_float_positional(self):
+        a = testing.shaped_arange((), cupy)
+        b = testing.shaped_arange((), numpy)
+        assert cupy.format_float_positional(a) == numpy.format_float_positional(
+            b
+        )
+
+    def test_format_float_scientific(self):
+        a = testing.shaped_arange((), cupy)
+        b = testing.shaped_arange((), numpy)
+        assert cupy.format_float_scientific(a) == numpy.format_float_scientific(
+            b
+        )

dpnp/tests/third_party/cupy/io_tests/test_npz.py

@@ -0,0 +1,104 @@
+import io
+import pickle
+import unittest
+
+import pytest
+
+import dpnp as cupy
+from dpnp.tests.third_party.cupy import testing
+
+pytest.skip("text functions are not supported yet", allow_module_level=True)
+
+
+class TestNpz(unittest.TestCase):
+
+    @testing.for_all_dtypes()
+    def test_save_load(self, dtype):
+        a = testing.shaped_arange((2, 3, 4), dtype=dtype)
+        sio = io.BytesIO()
+        cupy.save(sio, a)
+        s = sio.getvalue()
+        sio.close()
+
+        sio = io.BytesIO(s)
+        b = cupy.load(sio)
+        sio.close()
+
+        testing.assert_array_equal(a, b)
+
+    def test_save_pickle(self):
+        data = object()
+
+        sio = io.BytesIO()
+        with self.assertRaises(ValueError):
+            cupy.save(sio, data, allow_pickle=False)
+        sio.close()
+
+        sio = io.BytesIO()
+        cupy.save(sio, data, allow_pickle=True)
+        sio.close()
+
+    def test_load_pickle(self):
+        a = testing.shaped_arange((2, 3, 4), dtype=cupy.float32)
+
+        sio = io.BytesIO()
+        a.dump(sio)
+        s = sio.getvalue()
+        sio.close()
+
+        sio = io.BytesIO(s)
+        b = cupy.load(sio, allow_pickle=True)
+        testing.assert_array_equal(a, b)
+        sio.close()
+
+        sio = io.BytesIO(s)
+        with self.assertRaises(ValueError):
+            cupy.load(sio, allow_pickle=False)
+        sio.close()
+
+    @testing.for_all_dtypes()
+    def check_savez(self, savez, dtype):
+        a1 = testing.shaped_arange((2, 3, 4), dtype=dtype)
+        a2 = testing.shaped_arange((3, 4, 5), dtype=dtype)
+
+        sio = io.BytesIO()
+        savez(sio, a1, a2)
+        s = sio.getvalue()
+        sio.close()
+
+        sio = io.BytesIO(s)
+        with cupy.load(sio) as d:
+            b1 = d["arr_0"]
+            b2 = d["arr_1"]
+        sio.close()
+
+        testing.assert_array_equal(a1, b1)
+        testing.assert_array_equal(a2, b2)
+
+    def test_savez(self):
+        self.check_savez(cupy.savez)
+
+    def test_savez_compressed(self):
+        self.check_savez(cupy.savez_compressed)
+
+    @testing.for_all_dtypes()
+    def test_pickle(self, dtype):
+        a = testing.shaped_arange((2, 3, 4), dtype=dtype)
+        s = pickle.dumps(a)
+        b = pickle.loads(s)
+        testing.assert_array_equal(a, b)
+
+    @testing.for_all_dtypes()
+    def test_dump(self, dtype):
+        a = testing.shaped_arange((2, 3, 4), dtype=dtype)
+
+        sio = io.BytesIO()
+        a.dump(sio)
+        s = sio.getvalue()
+        sio.close()
+
+        sio = io.BytesIO(s)
+        b = cupy.load(sio, allow_pickle=True)
+        sio.close()
+
+        testing.assert_array_equal(a, b)

dpnp/tests/third_party/cupy/io_tests/test_text.py

@@ -0,0 +1,28 @@
+import filecmp
+import os
+import tempfile
+import unittest
+
+import numpy
+import pytest
+
+import dpnp as cupy
+
+pytest.skip("text functions are not supported yet", allow_module_level=True)
+
+
+class TestText(unittest.TestCase):
+
+    def test_savetxt(self):
+        tmp_cupy = tempfile.NamedTemporaryFile(delete=False)
+        tmp_numpy = tempfile.NamedTemporaryFile(delete=False)
+        try:
+            tmp_cupy.close()
+            tmp_numpy.close()
+            array = [[1, 2, 3], [2, 3, 4]]
+            cupy.savetxt(tmp_cupy.name, cupy.array(array))
+            numpy.savetxt(tmp_numpy.name, numpy.array(array))
+            assert filecmp.cmp(tmp_cupy.name, tmp_numpy.name)
+        finally:
+            os.remove(tmp_cupy.name)
+            os.remove(tmp_numpy.name)