Method write_read_atom_shape_args

tables/tests/test_array.py:158–256

(self, testarray)

            Path(filename).unlink()

    def write_read_atom_shape_args(self, testarray):
        a = testarray
        atom = tb.Atom.from_dtype(a.dtype)
        shape = a.shape
        byteorder = None

        if common.verbose:
            print("\n", "-=" * 30)
            print(
                "Running test for array with type '%s'" % a.dtype.type, end=" "
            )
            print("for class check:", self.title)

        # Create an instance of HDF5 file
        filename = tempfile.mktemp(".h5")
        try:
            with tb.open_file(filename, mode="w") as fileh:
                root = fileh.root

                # Create the array under root and name 'somearray'
                if self.endiancheck and a.dtype.kind != "S":
                    b = a.byteswap()
                    b.dtype = a.dtype.newbyteorder()
                    if b.dtype.byteorder in (">", "<"):
                        byteorder = tb.utils.byteorders[b.dtype.byteorder]
                    a = b

                ptarr = fileh.create_array(
                    root,
                    "somearray",
                    atom=atom,
                    shape=shape,
                    title="Some array",
                    # specify the byteorder explicitly
                    # since there is no way to deduce
                    # it in this case
                    byteorder=byteorder,
                )
                self.assertEqual(shape, ptarr.shape)
                self.assertEqual(atom, ptarr.atom)
                ptarr[...] = a

            # Re-open the file in read-only mode
            with tb.open_file(filename, mode="r") as fileh:
                root = fileh.root

                # Read the saved array
                b = root.somearray.read()

                # Compare them. They should be equal.
                if common.verbose and not common.allequal(a, b):
                    print("Write and read arrays differ!")
                    # print("Array written:", a)
                    print("Array written shape:", a.shape)
                    print("Array written itemsize:", a.itemsize)
                    print("Array written type:", a.dtype.type)
                    # print("Array read:", b)
                    print("Array read shape:", b.shape)