def test_query_starts_with_comment(): query = '# comment\nUSE test;' assert query_starts_with(query, ('use', )) is True
def test_query_starts_with(): query = 'USE test;' assert query_starts_with(query, ('use', )) is True query = 'DROP DATABASE test;' assert query_starts_with(query, ('use', )) is False