def test_query_where(self): expr = ops.index(self.table, self.col_i > 5) result = eval(expr) row1, row2 = result self.assertEqual((int(row1[0]), str(row1[1]), float(row1[2])), (8, "world", 4.2)) self.assertEqual((int(row2[0]), str(row2[1]), float(row2[2])), (16, "!", 8.4))
def test_select_where(self): expr = ops.index(self.col_i + 2 * self.col_price, blaze.logical_and(self.col_price > 5, self.col_price < 7)) result = eval(expr) np_result = (self.np_i + 2 * self.np_price)[ np.logical_and(self.np_price > 5, self.np_price < 7)] self.assertEqual([float(x) for x in result], [float(x) for x in np_result])
def test_query_where(self): expr = ops.index(self.col_i + self.col_i, self.col_i > 5) result = eval(expr) self.assertEqual([int(x) for x in result], [16, 32])
def test_index_sql_result_table(self): expr = ops.index(self.table, self.col_i > 5) result = eval(expr) i_col = result[:, 'i'] self.assertEqual([int(i_col[0]), int(i_col[1])], [8, 16])