mirror of
https://github.com/simonw/sqlite-utils.git
synced 2026-07-26 10:54:32 +02:00
69a1c0d960* Automated upgrades by Ruff
uvx --with 'ruff>=0.16.0' ruff check . --fix --unsafe-fixes
* Fix remaining Ruff errors with GPT-5.6 Sol high
https://gist.github.com/simonw/6da7906a9fea6e90da131c21a9055199
* Fix flake E501 long lines
* New Protocol for migrations to make ty happy
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import pytest
|
|
|
|
from sqlite_utils.utils import column_affinity
|
|
|
|
EXAMPLES = [
|
|
# Examples from https://www.sqlite.org/datatype3.html#affinity_name_examples
|
|
("INT", int),
|
|
("INTEGER", int),
|
|
("TINYINT", int),
|
|
("SMALLINT", int),
|
|
("MEDIUMINT", int),
|
|
("BIGINT", int),
|
|
("UNSIGNED BIG INT", int),
|
|
("INT2", int),
|
|
("INT8", int),
|
|
("CHARACTER(20)", str),
|
|
("VARCHAR(255)", str),
|
|
("VARYING CHARACTER(255)", str),
|
|
("NCHAR(55)", str),
|
|
("NATIVE CHARACTER(70)", str),
|
|
("NVARCHAR(100)", str),
|
|
("TEXT", str),
|
|
("CLOB", str),
|
|
("BLOB", bytes),
|
|
("REAL", float),
|
|
("DOUBLE", float),
|
|
("DOUBLE PRECISION", float),
|
|
("FLOAT", float),
|
|
# Numeric, treated as float:
|
|
("NUMERIC", float),
|
|
("DECIMAL(10,5)", float),
|
|
("BOOLEAN", float),
|
|
("DATE", float),
|
|
("DATETIME", float),
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize("column_def,expected_type", EXAMPLES)
|
|
def test_column_affinity(column_def, expected_type):
|
|
assert expected_type is column_affinity(column_def)
|
|
|
|
|
|
@pytest.mark.parametrize("column_def,expected_type", EXAMPLES)
|
|
def test_columns_dict(fresh_db, column_def, expected_type):
|
|
fresh_db.execute(f"create table foo (col {column_def})")
|
|
assert {"col": expected_type} == fresh_db["foo"].columns_dict
|