import unittest from block import __main__ class TestBlock(unittest.TestCase): def test_main_success(self): result = __main__(probability=0.2707298696) self.assertEqual(result, {"bin": "1"}) def test_main_invalid_input(self): with self.assertRaises(ValueError): __main__(probability="0.27") # Invalid input type (string) if __name__ == "__main__": unittest.main()