Function test_method

tables/tests/test_queries.py:416–505

(self)

    @ignore_skipped
    def test_method(self):
        common.verbosePrint("* Condition is ``%s``." % cond)
        # Replace bitwise operators with their logical counterparts.
        pycond = cond
        for ptop, pyop in [("&", "and"), ("|", "or"), ("~", "not")]:
            pycond = pycond.replace(ptop, pyop)
        pycond = compile(pycond, "<string>", "eval")

        table = self.table
        self.create_indexes(colname, ncolname, "c_idxextra")

        table_slice = dict(start=1, stop=table.nrows - 5, step=3)
        rownos, fvalues = None, None
        # Test that both simple and nested columns work as expected.
        # Knowing how the table is filled, results must be the same.
        for acolname in [colname, ncolname]:
            # First the reference Python version.
            pyrownos, pyfvalues, pyvars = [], [], condvars.copy()
            for row in table.iterrows(**table_slice):
                pyvars[colname] = row[acolname]
                pyvars["c_extra"] = row["c_extra"]
                pyvars["c_idxextra"] = row["c_idxextra"]
                try:
                    with warnings.catch_warnings():
                        warnings.filterwarnings(
                            "ignore",
                            "invalid value encountered in arc(cos|sin)",
                            RuntimeWarning,
                        )
                        isvalidrow = eval(pycond, func_info, pyvars)
                except TypeError:
                    raise SilentlySkipTest(
                        "The Python type does not support the operation."
                    )
                if isvalidrow:
                    pyrownos.append(row.nrow)
                    pyfvalues.append(row[acolname])
            pyrownos = np.array(pyrownos)  # row numbers already sorted
            pyfvalues = np.array(pyfvalues, dtype=sctype)
            pyfvalues.sort()
            common.verbosePrint(
                f"* {len(pyrownos)} rows selected by Python "
                f"from ``{acolname}``."
            )
            if rownos is None:
                rownos = pyrownos  # initialise reference results
                fvalues = pyfvalues
            else:
                self.assertTrue(np.all(pyrownos == rownos))  # check
                self.assertTrue(np.all(pyfvalues == fvalues))

            # Then the in-kernel or indexed version.
            ptvars = condvars.copy()
            ptvars[colname] = table.colinstances[acolname]
            ptvars["c_extra"] = table.colinstances["c_extra"]
            ptvars["c_idxextra"] = table.colinstances["c_idxextra"]
            try:
                isidxq = table.will_query_use_indexing(cond, ptvars)