Method test02a

tables/tests/test_nestedtypes.py:1528–1580

Indexing two simple columns under the same nested column.

(self)

        )

    def test02a(self):
        """Indexing two simple columns under the same nested column."""

        desc = {"nested": {"i1": tb.Int32Col(), "i2": tb.Int32Col()}}

        i1 = "nested/i1"
        i2 = "nested/i2"
        tbl = self.h5file.create_table(
            "/", "test", desc, title=self._getMethodName()
        )

        row = tbl.row
        for i in range(1000):
            row[i1] = i
            row[i2] = i * 2
            row.append()
        tbl.flush()

        cols = {
            "i1": tbl.cols.nested.i1,
            "i2": tbl.cols.nested.i2,
        }
        cols["i1"].create_index()
        cols["i2"].create_index()

        if self.reopen:
            self._reopen()
            tbl = self.h5file.root.test
            # Redefine the cols dictionary
            cols = {
                "i1": tbl.cols.nested.i1,
                "i2": tbl.cols.nested.i2,
            }

        i1res = [r[i1] for r in tbl.where("i1 < 10", cols)]
        i2res = [r[i2] for r in tbl.where("i2 < 10", cols)]

        if common.verbose:
            print("Retrieved values (i1):", i1res)
            print("Should look like:", list(range(10)))
            print("Retrieved values (i2):", i2res)
            print("Should look like:", list(range(0, 10, 2)))

        self.assertEqual(
            i1res,
            list(range(10)),
            "Select for nested column (i1) doesn't match.",
        )
        self.assertEqual(
            i2res,
            list(range(0, 10, 2)),
            "Select for nested column (i2) doesn't match.",
        )

    def test02b(self):
        """Indexing two simple columns under the same (very) nested column."""