import unittest
from block import __main__

class TestBlock(unittest.TestCase):

    def test_main_success(self):
        result = __main__(record_counts_negative_trade_count=7, record_counts_installment_trade_count=8,
                          record_counts_total_trade_count=18, record_counts_total_inquiry_count=3,
                          record_counts_revolving_trade_count=9, score_results=600.0,
                          installment_amount_monthly_payment=572, revolving_amount_percent_available_credit=18,
                          G069S=3, AT24S=6, BR02S=1, BI02S=5, AGG103=27642.0, ALL231=-2.0, AT12S=7, IN20S=166,
                          AT33A=38353, AT35A=5479, AT28A=54087, AT34B=71, S061S=1, RE102S=2000)

        expected_result = {'score_results': 600.0, 'AT34B': 71, 'AT12S': 7, 'revolving_amount_percent_available_credit': 18, 'AT28A': 54087, 'record_counts_total_trade_count': 18, 'record_counts_negative_trade_count': 7, 'record_counts_revolving_trade_count': 9, 'AT33A': 38353, 'AT35A': 5479, 'record_counts_total_inquiry_count': 3, 'IN20S': 150, 'RE102S': 2000, 'installment_amount_monthly_payment': 572, 'S061S': 1, 'record_counts_installment_trade_count': 8, 'BR02S': 1, 'AGG103': 27642.0, 'ALL231': 0.0, 'G069S': 3, 'AT24S': 6, 'BI02S': 3}
        
        for key, expected_value in expected_result.items():
            if isinstance(expected_value, float):
                self.assertAlmostEqual(result[key], expected_value, places=6, msg=f"Mismatch for {key}")
            else:
                self.assertEqual(result[key], expected_value, msg=f"Mismatch for {key}")

    # def test_main_invalid_input(self):
    #     with self.assertRaises(TypeError):
    #         __main__(record_counts_negative_trade_count=7, record_counts_installment_trade_count=8,
    #                  record_counts_total_trade_count=18, record_counts_total_inquiry_count=3,
    #                  record_counts_revolving_trade_count=9, score_results=600, installment_amount_monthly_payment=572,
    #                  revolving_amount_percent_available_credit=18, G069S=3, AT24S=6, BR02S=1, BI02S=5,
    #                  AGG103=27642, ALL231=-2, AT12S=7, IN20S=166, AT33A=38353, AT35A=5479, AT28A=54087,
    #                  AT34B=71, S061S=1, RE102S=2000)

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