import unittest
from block import __main__ 

data = {'pd_a': 0.030282551422715187, 'pd_b': 0.07098247110843658, 'pd_t': 0.6349245309829712}

class TestBlock(unittest.TestCase):
    def test_main_returns_scores(self):
        block_result = __main__(**data)
        print(block_result)
        self.assertIsInstance(block_result, dict)
        self.assertIn("pd_a", block_result)
        self.assertIn("pd_b", block_result)
        self.assertIn("pd_t", block_result)
        self.assertIn("pd_iso_a", block_result)
        self.assertIn("pd_iso_b", block_result)
        self.assertIn("final_pd", block_result)
        self.assertIn("grade", block_result)

if __name__ == "__main__":  # pragma: no cover
    unittest.main()