Skip to content

Commit b6388a3

Browse files
author
Ryan Delaney
committed
Add tests for get_pass1_budget()
1 parent 93cb8c8 commit b6388a3

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

tests/test_functions.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -383,7 +383,14 @@ class test_add_acpl(unittest.TestCase):
383383

384384

385385
class test_get_pass1_budget(unittest.TestCase):
386-
pass
386+
387+
def test_raises_typeerror(self):
388+
self.assertRaises(TypeError, annotator.get_pass1_budget, 'a')
389+
390+
def test_math(self):
391+
seed = random.random()
392+
result = annotator.get_pass1_budget(seed)
393+
self.assertEqual(result, seed / 10)
387394

388395

389396
class test_get_pass2_budget(unittest.TestCase):

0 commit comments

Comments
 (0)