Method test09b

tables/tests/test_tables.py:5488–5530

Checking modifying table columns (multiple columns, recarray)

(self)

        self.assertEqual(table.nrows, 4)

    def test09b(self):
        """Checking modifying table columns (multiple columns, recarray)"""

        if common.verbose:
            print("\n", "-=" * 30)
            print("Running %s.test09b..." % self.__class__.__name__)

        # Create a new table:
        table = self.h5file.create_table(self.h5file.root, "recarray", Rec)

        # append new rows
        r = np.rec.array(
            [(456, b"dbe", 1.2), (2, b"ded", 1.3)], formats="i4,S3,f8"
        )
        table.append(r)
        table.append([(457, b"db1", 1.2), (5, b"de1", 1.3)])

        # Modify a couple of columns
        columns = np.rec.array(
            [("aaa", 1.2), ("bbb", 0.1), ("ccc", 0.3)], formats="S3,f8"
        )
        table.modify_columns(start=1, columns=columns, names=["col2", "col3"])
        # Create the modified recarray
        r1 = np.rec.array(
            [
                (456, "dbe", 1.2),
                (2, "aaa", 1.2),
                (457, "bbb", 0.1),
                (5, "ccc", 0.3),
            ],
            formats="i4,S3,f8",
            names="col1,col2,col3",
        )
        # Read the modified table
        if self.reopen:
            self._reopen()
            table = self.h5file.root.recarray
        r2 = table.read()
        if common.verbose:
            print("Original table-->", repr(r2))
            print("Should look like-->", repr(r1))
        self.assertEqual(r1.tobytes(), r2.tobytes())
        self.assertEqual(table.nrows, 4)

    def test09c(self):
        """Checking modifying table columns (single column, step)"""