From d4a81c11d2605344cc300b68618f897f5054f855 Mon Sep 17 00:00:00 2001 From: t_max <1172915550@qq.com> Date: Thu, 23 Mar 2023 14:07:43 +0800 Subject: [PATCH] fix: fix check timestamp data --- tests/pytest/util/sql.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/pytest/util/sql.py b/tests/pytest/util/sql.py index 5ba96de24c..414b8b2099 100644 --- a/tests/pytest/util/sql.py +++ b/tests/pytest/util/sql.py @@ -285,13 +285,13 @@ class TDSql: success = False if precision == 'ms': dt_obj = self.queryResult[row][col] - tz = int(int((dt_obj-datetime.datetime.fromtimestamp(0,dt_obj.tzinfo)).total_seconds())*1000) + int(dt_obj.microsecond/1000) - if tz == data: + ts = int(int((dt_obj-datetime.datetime.fromtimestamp(0,dt_obj.tzinfo)).total_seconds())*1000) + int(dt_obj.microsecond/1000) + if ts == data: success = True elif precision == 'us': dt_obj = self.queryResult[row][col] - tz = int(int((dt_obj-datetime.datetime.fromtimestamp(0,dt_obj.tzinfo)).total_seconds())*1e6) + int(dt_obj.microsecond) - if tz == data: + ts = int(int((dt_obj-datetime.datetime.fromtimestamp(0,dt_obj.tzinfo)).total_seconds())*1e6) + int(dt_obj.microsecond) + if ts == data: success = True elif precision == 'ns': if data == self.queryResult[row][col]: