Antibody — sqlparse

Catching tests:

New test:

import timeit
import sqlparse


def test_reindent_tuple_list_not_quadratic():
    def make_sql(n):
        values = ", ".join(f"({i}, {i+1})" for i in range(n))
        return f"SELECT a, b FROM t WHERE (a, b) IN ({values})"

    def time_op(n):
        sql = make_sql(n)
        return min(timeit.repeat(lambda: sqlparse.format(sql, reindent=True), number=1, repeat=3))

    n = 100
    t_n = time_op(n)
    t_4n = time_op(n * 4)
    ratio = t_4n / t_n
    assert ratio < 9, f"growth ratio {ratio:.1f} suggests quadratic behaviour"

Catching tests:

New test:

import timeit
import sqlparse

def test_strip_comments_not_quadratic():
    def make_sql(n):
        return "\n".join(f"-- comment {i}" for i in range(n))

    def time_op(n):
        sql = make_sql(n)
        return min(timeit.repeat(lambda: sqlparse.format(sql, strip_comments=True), number=1, repeat=3))

    n = 500
    t_n = time_op(n)
    t_4n = time_op(n * 4)
    ratio = t_4n / t_n
    assert ratio < 9, f"growth ratio {ratio:.1f} suggests quadratic behaviour"

Catching tests:

Catching tests:

Catching tests:

Catching tests:

Catching tests:

Catching tests:

Catching tests:

New test:

import sqlparse
from sqlparse import tokens as T


def test_attach_detach_are_keywords():
    # ATTACH and DETACH should be recognized as Keyword tokens in PostgreSQL dialect
    for word, sql in [
        ("DETACH", "ALTER TABLE atable DETACH PARTITION atable_p00"),
        ("ATTACH", "ALTER INDEX atable_acolumn_idx ATTACH PARTITION atable_p01_acolumn_idx"),
    ]:
        parsed = sqlparse.parse(sql)[0]
        found_type = None
        for token in parsed.flatten():
            if token.normalized.upper() == word:
                found_type = token.ttype
                break
        assert found_type == T.Keyword, (
            f"Expected {word} to have token type {T.Keyword}, got {found_type}"
        )

Catching tests:

New test:

import sqlparse
from sqlparse import tokens as T


def test_extension_tokenized_as_keyword():
    """EXTENSION should be tokenized as a Keyword, not part of an Identifier."""
    stmt = "DROP EXTENSION pgcrypto"
    parsed = sqlparse.parse(stmt)[0]

    extension_token = None
    for token in parsed.flatten():
        if token.normalized.upper() == 'EXTENSION':
            extension_token = token
            break

    assert extension_token is not None, "EXTENSION token not found in parsed statement"
    assert extension_token.ttype is T.Keyword, (
        f"Expected EXTENSION to have ttype {T.Keyword}, got {extension_token.ttype}"
    )

Catching tests:

Catching tests:

Catching tests:

Catching tests:

New test:

import sqlparse
from sqlparse import tokens as T


def test_truncate_ddl_grant_revoke_dcl():
    """TRUNCATE should be Keyword.DDL; GRANT and REVOKE should be Keyword.DCL."""
    def first_ttype(sql):
        return sqlparse.parse(sql)[0].tokens[0].ttype

    truncate_ttype = first_ttype("TRUNCATE")
    grant_ttype = first_ttype("GRANT")
    revoke_ttype = first_ttype("REVOKE")

    assert truncate_ttype == T.Keyword.DDL, (
        f"TRUNCATE ttype is {truncate_ttype}, expected {T.Keyword.DDL}"
    )
    assert grant_ttype == T.Keyword.DCL, (
        f"GRANT ttype is {grant_ttype}, expected {T.Keyword.DCL}"
    )
    assert revoke_ttype == T.Keyword.DCL, (
        f"REVOKE ttype is {revoke_ttype}, expected {T.Keyword.DCL}"
    )

Probe: .antibody/sqlparse/checks/8c24779e02.py
With fix:

DIFFERENCE FOUND: indents=[0, 7] vs all-equal

DIFFERENCE FOUND: indents=[0, 7] vs all-equal

Catching tests:

Catching tests:

Catching tests:

Catching tests:

Note: INCONCLUSIVE (data-only change)