Method setUp

tables/tests/test_expression.py:196–270

(self)

class MixedContainersTestCase(common.TempFileMixin, common.PyTablesTestCase):

    def setUp(self):
        super().setUp()

        # The expression
        self.expr = "2 * a*b + c**2+d**2+e-f+g"

        # Create a directory in file for outputs
        root = self.h5file.root
        outs = self.h5file.create_group(root, "outs")

        # Define the NumPy variables to be used in expression
        N = np.prod(self.shape)

        # Initial values for variables
        a = np.arange(0, N, dtype="int32").reshape(self.shape)
        b = np.arange(N, 2 * N, dtype="int64").reshape(self.shape)
        c = np.arange(2 * N, 3 * N, dtype="int32").reshape(self.shape)
        d = np.arange(3 * N, 4 * N, dtype="int32").reshape(self.shape)
        e = np.arange(4 * N, 5 * N, dtype="int32").reshape(self.shape)
        self.f = f = int(3)  # a regular python type
        self.g = g = np.int16(2)  # a NumPy scalar type

        # Original values
        self.npvars = {"a": a, "b": b, "c": c, "d": d, "e": e, "f": f, "g": g}
        rnda = b.copy()

        # ndarray input and output
        self.a = a
        self.rnda = rnda

        # Array input and output
        self.b = self.h5file.create_array(root, "b", b)
        self.rarr = self.b.copy(outs)

        # CArray input and output
        self.c = self.h5file.create_carray(
            root, "c", atom=tb.Atom.from_dtype(c.dtype), shape=self.shape
        )
        self.c[:] = c
        self.rcarr = self.c.copy(outs)

        # EArray input and output
        eshape = list(self.shape)
        eshape[0] = 0
        self.d = self.h5file.create_earray(
            root, "d", atom=tb.Atom.from_dtype(d.dtype), shape=eshape
        )
        self.d.append(d)
        self.rearr = self.d.copy(outs)

        # Column input and output
        rtype = {}
        colshape = self.shape[1:]
        for i, col in enumerate((a, b, c, d, e, rnda)):
            rtype["f%d" % i] = tb.Col.from_sctype(col.dtype.type, colshape)
        t = self.h5file.create_table(root, "t", rtype)
        nrows = self.shape[0]
        row = t.row