diff options
| -rw-r--r-- | ydb/core/kqp/ut/arrow/kqp_arrow_in_channels_ut.cpp | 2 | ||||
| -rw-r--r-- | ydb/core/kqp/ut/arrow/kqp_types_arrow_ut.cpp | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/ydb/core/kqp/ut/arrow/kqp_arrow_in_channels_ut.cpp b/ydb/core/kqp/ut/arrow/kqp_arrow_in_channels_ut.cpp index 4209f59e66c..51e2bfbbb7b 100644 --- a/ydb/core/kqp/ut/arrow/kqp_arrow_in_channels_ut.cpp +++ b/ydb/core/kqp/ut/arrow/kqp_arrow_in_channels_ut.cpp @@ -85,7 +85,7 @@ void InsertAllColumnsAndCheckSelectAll(TKikimrRunner* runner) { UNIT_ASSERT(*parser.ColumnParser(17).GetOptionalJsonDocument().Get() == TString("[14]")); UNIT_ASSERT(*parser.ColumnParser(18).GetOptionalDyNumber().Get() == TString(".1515e2")); streamPart = it.ReadNext().GetValueSync(); - UNIT_ASSERT_C(streamPart.EOS(), streamPart.GetIssues().ToString()); + UNIT_ASSERT_C(streamPart.EOS() || streamPart.HasQueryStats(), streamPart.GetIssues().ToString()); } } diff --git a/ydb/core/kqp/ut/arrow/kqp_types_arrow_ut.cpp b/ydb/core/kqp/ut/arrow/kqp_types_arrow_ut.cpp index fd834e6874b..d5abf13d62e 100644 --- a/ydb/core/kqp/ut/arrow/kqp_types_arrow_ut.cpp +++ b/ydb/core/kqp/ut/arrow/kqp_types_arrow_ut.cpp @@ -91,7 +91,7 @@ void InsertAllColumnsAndCheckSelectAll(TKikimrRunner* runner) { UNIT_ASSERT(*parser.ColumnParser(17).GetOptionalJsonDocument().Get() == TString("[14]")); UNIT_ASSERT(*parser.ColumnParser(18).GetOptionalDyNumber().Get() == TString(".1515e2")); streamPart = it.ReadNext().GetValueSync(); - UNIT_ASSERT_C(streamPart.EOS(), streamPart.GetIssues().ToString()); + UNIT_ASSERT_C(streamPart.EOS() || streamPart.HasQueryStats(), streamPart.GetIssues().ToString()); } } |