test_base_column.py

import unittest

import decimal

from dbt.adapters.base import Column


class TestStringType(unittest.TestCase):
    def test__character_type(self):
        col = Column("fieldname", "character", char_size=10)

        self.assertEqual(col.data_type, "character varying(10)")


class TestNumericType(unittest.TestCase):
    def test__numeric_type(self):
        col = Column(
            "fieldname",
            "numeric",
            numeric_precision=decimal.Decimal("12"),
            numeric_scale=decimal.Decimal("2"),
        )

        self.assertEqual(col.data_type, "numeric(12,2)")

    def test__numeric_type_with_no_precision(self):
        # PostgreSQL, at least, will allow empty numeric precision
        col = Column("fieldname", "numeric", numeric_precision=None)

        self.assertEqual(col.data_type, "numeric")