Method test02b

tables/tests/test_nestedtypes.py:1582–1641

Indexing two simple columns under the same (very) nested column.

(self)

        )

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

        desc = {
            "nested1": {
                "nested2": {
                    "nested3": {"i1": tb.Int32Col(), "i2": tb.Int32Col()}
                }
            }
        }

        i1 = "nested1/nested2/nested3/i1"
        i2 = "nested1/nested2/nested3/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.nested1.nested2.nested3.i1,
            "i2": tbl.cols.nested1.nested2.nested3.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.nested1.nested2.nested3.i1,
                "i2": tbl.cols.nested1.nested2.nested3.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)),