summaryrefslogtreecommitdiff
path: root/test/ext/mypy/plugin_files/boolean_col.py
blob: 3e361ad10223d041695dab81e8aaf2c0af1f75a9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from typing import Optional

from sqlalchemy import Boolean
from sqlalchemy import Column
from sqlalchemy.orm import declarative_base

Base = declarative_base()


class TestBoolean(Base):
    __tablename__ = "test_boolean"

    flag = Column(Boolean)

    bflag: bool = Column(Boolean(create_constraint=True))


expr = TestBoolean.flag.is_(True)

t1 = TestBoolean(flag=True)

x: Optional[bool] = t1.flag

y: bool = t1.bflag