16 lines
426 B
Python
16 lines
426 B
Python
|
import unittest
|
||
|
|
from block import __main__
|
||
|
|
|
||
|
|
class TestBlock(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_main_success(self):
|
||
|
|
result = __main__(probability=0.3890174329280853)
|
||
|
|
self.assertEqual(result, {"grade": "G2"})
|
||
|
|
|
||
|
|
def test_main_invalid_input(self):
|
||
|
|
with self.assertRaises(ValueError):
|
||
|
|
__main__(probability="0.40") # Invalid input type (string)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|