xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

decimal operator tests

This commit is contained in:
wangjiaming0909 2025-02-26 23:01:10 +08:00
parent afa6301e32
commit ba4b30c0f3
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tests/system-test/2-query/decimal.py
View File

@ -1,6 +1,8 @@
from ast import Tuple from ast import Tuple
import math
from pydoc import doc from pydoc import doc
from random import randrange from random import randrange
import random
from re import A from re import A
import time import time
import threading import threading
@ -159,7 +161,7 @@ class TaosShell:
lines = lines[1:] lines = lines[1:]
for line in lines: for line in lines:
col = 0 col = 0
vals: List[str] = line.split(",") vals: list[str] = line.split(",")
if len(self.queryResult) == 0: if len(self.queryResult) == 0:
self.queryResult = [[] for i in range(len(vals))] self.queryResult = [[] for i in range(len(vals))]
for val in vals: for val in vals:
@ -184,10 +186,10 @@ class DecimalColumnExpr:
def __init__(self, format: str, executor): def __init__(self, format: str, executor):
self.format_: str = format self.format_: str = format
self.executor_ = executor self.executor_ = executor
self.params_: Tuple = List self.params_: Tuple = ()
def __str__(self): def __str__(self):
return self.format_ % (self.params_) return f"({self.format_})".format(*self.params_)
def execute(self, params): def execute(self, params):
return self.executor_(params) return self.executor_(params)
@ -213,14 +215,16 @@ class DecimalColumnExpr:
tdLog.exit(f"query with expr: {self} calc in py got: {v_from_calc_in_py}, query got: {v_from_query}") tdLog.exit(f"query with expr: {self} calc in py got: {v_from_calc_in_py}, query got: {v_from_query}")
tdLog.debug(f"query with expr: {self} calc got same result: NULL") tdLog.debug(f"query with expr: {self} calc got same result: NULL")
continue continue
failed = False
if isinstance(v_from_calc_in_py, float): if isinstance(v_from_calc_in_py, float):
dec_from_query = float(v_from_query) dec_from_query = float(v_from_query)
dec_from_insert = float(v_from_calc_in_py) dec_from_insert = float(v_from_calc_in_py)
failed = not math.isclose(dec_from_query, dec_from_insert, rel_tol=1e-9, abs_tol=1e-9)
else: else:
dec_from_query = Decimal(v_from_query) dec_from_query = Decimal(v_from_query)
dec_from_insert = Decimal(v_from_calc_in_py) dec_from_insert = Decimal(v_from_calc_in_py)
if dec_from_query != dec_from_insert: failed = dec_from_query != dec_from_insert
if failed:
tdLog.exit( tdLog.exit(
f"check decimal column failed for expr: {self} params: {params}, query: {v_from_query}, expect {dec_from_insert}, but get {dec_from_query}") f"check decimal column failed for expr: {self} params: {params}, query: {v_from_query}, expect {dec_from_insert}, but get {dec_from_query}")
else: else:
@ -231,7 +235,7 @@ class DecimalColumnExpr:
def generate(self, format_params) -> str: def generate(self, format_params) -> str:
self.params_ = format_params self.params_ = format_params
return f"({self.format_})".format(*format_params) return self.__str__()
class TypeEnum: class TypeEnum:
@ -486,7 +490,7 @@ class Column:
def __str__(self): def __str__(self):
if self.is_constant_col(): if self.is_constant_col():
return self.get_constant_val() return str(self.get_constant_val())
return self.name_ return self.name_
def get_val(self, tbname: str, idx: int): def get_val(self, tbname: str, idx: int):
@ -1184,11 +1188,9 @@ class TDTestCase:
if not unsupported_col.type_.type in unsupported_type: if not unsupported_col.type_.type in unsupported_type:
continue continue
for binary_op in DecimalBinaryOperator.get_all_binary_ops(): for binary_op in DecimalBinaryOperator.get_all_binary_ops():
select_expr = binary_op.generate((col.name_, unsupported_col.name_)) select_expr = binary_op.generate((col, unsupported_col))
sql = f"select {select_expr} from {self.db_name}.{tbname}" sql = f"select {select_expr} from {self.db_name}.{tbname}"
select_expr_reverse = binary_op.generate( select_expr_reverse = binary_op.generate((unsupported_col, col))
(unsupported_col.name_, col.name_)
)
sql_reverse = ( sql_reverse = (
f"select {select_expr_reverse} from {self.db_name}.{tbname}" f"select {select_expr_reverse} from {self.db_name}.{tbname}"
) )
@ -1212,7 +1214,7 @@ class TDTestCase:
## tables: meters, nt ## tables: meters, nt
## columns: c1, c2, c3, c4, c5, c7, c8, c9, c10, c99, c100 ## columns: c1, c2, c3, c4, c5, c7, c8, c9, c10, c99, c100
binary_operators = [ binary_operators = [
DecimalColumnExpr("%s + %s", DecimalBinaryOperator.execute_plus), DecimalColumnExpr("{0} + {1}", DecimalBinaryOperator.execute_plus),
# DecimalColumnExpr("-"), # DecimalColumnExpr("-"),
# DecimalColumnExpr("*"), # DecimalColumnExpr("*"),
# DecimalColumnExpr("/"), # DecimalColumnExpr("/"),