def METHOD_NAME(self):
    test_table = self.session.time_table(period=1000000)
    while test_table.snapshot().size < 100:
        time.sleep(0.001)
    query = self.session.query(test_table)
    (query.update(formulas=["Col1 = i", "Col2 = i * 2"])
     .where(["Col1 > 10"])
     .snapshot()
     .head(10))
    result_table = query.exec()
    self.assertEqual(result_table.to_arrow().num_rows, 10)