re.split for data queries

Author: dmitrii-ubskii

test/code/runners/typeql_runner.py

@@ -3,6 +3,7 @@
 from enum import Enum
 from typing import List, Dict, Tuple, Union
 from test.code.parser.parser import ParsedTest
+import re
 import logging
 logger = logging.getLogger('main')
 # To see debug log, set logging level to debug in main.py
@@ -93,7 +94,7 @@ def run_failing_queries(self, queries: List[str], type: TransactionType) -> str:
         return FailureMode.NoFailure
 
     def run_transaction(self, queries: List[str], type: TransactionType, rollback=False) -> Union[int, None]:
-        queries = [q for qs in queries for q in qs.split('end;') if q.strip()]
+        queries = [q for qs in queries for q in re.split(r'end\s*;', qs) if q.strip()]
         with self.driver.transaction(self.db, type) as tx:
             try:
                 for q in queries: