test_bahttext.py

# -*- coding: utf-8 -*-
# Convert float decimal number to Thai text number
# Copyright (C) 2014, Morange Solution Co.,LTD.
# This file is distributed under the same license as the bahttext package.
# All rights reserved.
# Seksan Poltree <seksan@morange.co.th>, 2014.

import unittest
from bahttext import bahttext


class TestBahtText(unittest.TestCase):

    def test_input_is_a_number_then_can_convert_to_string(self):
        self.assertIsInstance(bahttext(1.0), str)

    def test_input_is_not_number_then_raise_error(self):
        with self.assertRaises(TypeError) as ex:
            bahttext('text input')
        self.assertIsInstance(ex.exception, TypeError)

    def test_input_integer_one_digit_can_convert(self):
        self.assertEqual(bahttext(0.0), 'ศูนย์บาทถ้วน')
        self.assertEqual(bahttext(1.0), 'หนึ่งบาทถ้วน')
        self.assertEqual(bahttext(2.0), 'สองบาทถ้วน')
        self.assertEqual(bahttext(5.0), 'ห้าบาทถ้วน')

    def test_number_in_multiple_could_show_sip(self):
        self.assertEqual(bahttext(10.0), 'สิบบาทถ้วน')
        self.assertEqual(bahttext(20.0), 'ยี่สิบบาทถ้วน')
        self.assertEqual(bahttext(50.0), 'ห้าสิบบาทถ้วน')
        self.assertEqual(bahttext(11.0), 'สิบเอ็ดบาทถ้วน')
        self.assertEqual(bahttext(15.0), 'สิบห้าบาทถ้วน')

    def test_input_one_integer_wth_decimal_can_convert(self):
        self.assertEqual(bahttext(1.10), 'หนึ่งบาทสิบสตางค์')
        self.assertEqual(bahttext(2.50), 'สองบาทห้าสิบสตางค์')
        self.assertEqual(bahttext(3.75), 'สามบาทเจ็ดสิบห้าสตางค์')

    def test_number_ending_with_one_could_show_edd(self):
        self.assertEqual(bahttext(4.81), 'สี่บาทแปดสิบเอ็ดสตางค์')

    def test_number_start_with_two_could_show_yee(self):
        self.assertEqual(bahttext(2.21), 'สองบาทยี่สิบเอ็ดสตางค์')

    def test_number_hundred_should_show_roi(self):
        self.assertEqual(bahttext(100.0), 'หนึ่งร้อยบาทถ้วน')
        self.assertEqual(bahttext(101.0), 'หนึ่งร้อยเอ็ดบาทถ้วน')
        self.assertEqual(bahttext(200.0), 'สองร้อยบาทถ้วน')
        self.assertEqual(bahttext(201.0), 'สองร้อยเอ็ดบาทถ้วน')

    def test_number_thousand_should_show_pan(self):
        self.assertEqual(bahttext(1000.0), 'หนึ่งพันบาทถ้วน')
        self.assertEqual(bahttext(2000.10), 'สองพันบาทสิบสตางค์')
        self.assertEqual(
            bahttext(3211.51), 'สามพันสองร้อยสิบเอ็ดบาทห้าสิบเอ็ดสตางค์')
        self.assertEqual(bahttext(8000.31), 'แปดพันบาทสามสิบเอ็ดสตางค์')

    def test_number_ten_thousand_should_show_muern(self):
        self.assertEqual(bahttext(30000.0), 'สามหมื่นบาทถ้วน')
        self.assertEqual(
            bahttext(98765.10), 'เก้าหมื่นแปดพันเจ็ดร้อยหกสิบห้าบาทสิบสตางค์')
        self.assertEqual(
            bahttext(30211.21), 'สามหมื่นสองร้อยสิบเอ็ดบาทยี่สิบเอ็ดสตางค์')

    def test_number_hundred_thousand_should_show_saan(self):
        self.assertEqual(bahttext(800000.0), 'แปดแสนบาทถ้วน')
        self.assertEqual(
            bahttext(258065.81), 'สองแสนห้าหมื่นแปดพันหกสิบห้าบาทแปดสิบเอ็ดสตางค์')

    def test_number_million_should_show_laan(self):
        self.assertEqual(bahttext(3500000.0), 'สามล้านห้าแสนบาทถ้วน')

    def test_number_multiple_million_should_show_multiple_laan(self):
        self.assertEqual(bahttext(12000000.0), 'สิบสองล้านบาทถ้วน')
        self.assertEqual(bahttext(21000000.0), 'ยี่สิบเอ็ดล้านบาทถ้วน')
        self.assertEqual(
            bahttext(51000000000000.51), 'ห้าสิบเอ็ดล้านล้านบาทห้าสิบเอ็ดสตางค์')
        self.assertEqual(
            bahttext(10000000680000.51), 'สิบล้านล้านหกแสนแปดหมื่นบาทห้าสิบเอ็ดสตางค์')

    def test_negative_minus_prefix_number_should_print_loub(self):
        self.assertEqual(bahttext(-1.10), 'ลบหนึ่งบาทสิบสตางค์')
        self.assertEqual(bahttext(-1000.0), 'ลบหนึ่งพันบาทถ้วน')
        self.assertEqual(
            bahttext(-258065.81), 'ลบสองแสนห้าหมื่นแปดพันหกสิบห้าบาทแปดสิบเอ็ดสตางค์'
        )

    def test_fraction_number_should_not_have_tuan(self):
        self.assertEqual(bahttext(12000000.0), 'สิบสองล้านบาทถ้วน')
        self.assertNotEqual(bahttext(12000000.0), 'สิบสองล้านบาทสตางค์')


if __name__ == '__main__':
    unittest.main()