CTAS: fix make (#22563)

Author: nikvas0

ydb/tests/stress/ctas/workload/__init__.py

@@ -1,9 +1,7 @@
 # -*- coding: utf-8 -*-
 import ydb
 import time
-import random
 import threading
-from enum import Enum
 
 from ydb.tests.stress.common.common import WorkloadBase
 
@@ -23,7 +21,7 @@ def __init__(self, client, prefix, stop, is_olap):
     def get_stat(self):
         with self.lock:
             return f"Tables: {self.tables_count}, TotalRows: {self.total_rows}, LastTableRows: {self.prev_rows[0]}"
-    
+
     def get_create_query(self):
         table_path = self.get_table_path(self.table_name + str(self.tables_count))
 
@@ -32,7 +30,7 @@ def get_create_query(self):
             params = "STORE = COLUMN"
         else:
             params = "STORE = ROW"
-        
+
         if self.tables_count == 0:
             return f"""
                 CREATE TABLE `{table_path}` (
@@ -51,9 +49,9 @@ def get_create_query(self):
             WITH (
                 {params}
             ) AS SELECT
-                Unwrap(CAST(1 AS Uint64)) AS Key, "{ ('0' * self.value_bytes) }" AS Value;
+                Unwrap(CAST(1 AS Uint64)) AS Key, "{('0' * self.value_bytes)}" AS Value;
         """
-    
+
     def get_ctas_query(self):
         table_path = self.get_table_path(self.table_name + str(self.tables_count))
         prev_table_path1 = self.get_table_path(self.table_name + str(self.tables_count - 1))
@@ -66,10 +64,10 @@ def get_ctas_query(self):
             FROM `{prev_table_path1}`
             UNION ALL
             SELECT
-                Key + Unwrap(CAST({self.prev_rows[0]} AS Uint64)) AS Key, Value 
+                Key + Unwrap(CAST({self.prev_rows[0]} AS Uint64)) AS Key, Value
             FROM `{prev_table_path2}`
         """
-    
+
     def prepare(self):
         create_query = self.get_create_query()
         self.client.query(
@@ -84,7 +82,6 @@ def prepare(self):
         )
         self.tables_count += 1
 
-
     def _loop(self):
         self.prepare()
 
@@ -105,7 +102,7 @@ def _loop(self):
             expected = self.prev_rows[0] + self.prev_rows[1]
             if actual != expected:
                 raise Exception(f"Incorrect result: expected:{expected}, actual:{actual}")
-            
+
             self.prev_rows = (actual, self.prev_rows[0])
             self.total_rows += actual
             self.tables_count += 1

ydb/tests/stress/ya.make

@@ -1,6 +1,7 @@
 RECURSE(
     cdc
     common
+    ctas
     kv
     log
     mixedpy