from sqlobject import *
from sqlobject.tests.dbtest import *

try:
    from decimal import Decimal
except ImportError:
    Decimal = None

########################################
## Decimal columns
########################################

class DecimalTable(SQLObject):
    col1 = DecimalCol(size=6, precision=4)

def test_1decimal():
    if not supports('decimalColumn'):
        return
    setupClass(DecimalTable)
    d = DecimalTable(col1=21.12)
    assert float(d.col1) == 21.12

if Decimal and supports('decimalColumn'):
    def test_2Decimal():
        setupClass(DecimalTable)
        d = DecimalTable(col1=Decimal("21.12"))
        if isinstance(d.col1, Decimal):
            assert d.col1 == Decimal("21.12")
        else:
            assert d.col1 == 21.12