23 lines
726 B
Python
23 lines
726 B
Python
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()
|
|
|
|
|