def test_row_count(unittest, table_name, count, condition=''):
test_sql = "SELECT * FROM " + table_name
if condition:
test_sql += ' WHERE ' + condition
result = exec_sql(test_sql)
unittest.assertEqual(len(result), count, table_name)
def get_row_count(unittest, table_name, count):
test_sql = "SELECT * FROM " + table_name
result = exec_sql(test_sql)
unittest.assertEqual(len(result), count, table_name)
def get_row_count(unittest, table_name, condition=''):
test_sql = "SELECT * FROM " + table_name
if condition:
test_sql += ' WHERE ' + condition
result = exec_sql(test_sql)
return len(result)
