tests.py

import unittest
from pathfinder_dice import Dice, roll


class TestAlgos(unittest.TestCase):

    def test_one_dice(self):
        with self.subTest():
            self.assertEqual(roll('1d1'), 1)
        with self.subTest():
            for i in range(20):
                self.assertIn(roll('1d2'), range(1, 3))
        with self.subTest('test dice without count'):
            self.assertEqual(roll('d1'), 1)
        with self.subTest():
            for i in range(20):
                self.assertIn(roll('d2'), range(1, 3))
        with self.subTest('test more than one digit in dice maximum value'):
            for i in range(100):
                self.assertIn(roll('d12'), range(1, 13))
        with self.subTest('test: return different results'):
            self.assertNotEqual(roll('d1000'), roll('d1000'))
        with self.subTest('test procent dice'):
            for i in range(100):
                self.assertIn(roll('d%'), range(1, 101))

    def test_two_dices(self):
        with self.subTest('test: two dices'):
            for i in range(20):
                self.assertIn(roll('2d2'), range(2, 5))
        with self.subTest('test: two dices != one dice * count'):
            for k in range(2, 5):
                self.assertIn(k, [roll('2d2') for i in range(20)])

    def test_many_dices(self):
        with self.subTest():
            self.assertEqual(roll('20d1'), 20)
        with self.subTest():
            for i in range(1000):
                self.assertIn(roll('20d2'), range(20, 40))

    def test_multiply(self):
        with self.subTest():
            self.assertEqual(roll('d1x2'), 2)
        with self.subTest():
            self.assertEqual(roll('d1x2x2'), 3)
        with self.subTest():
            self.assertEqual(roll('d1x2x3x4'), 7)

    def test_math(self):
        with self.subTest('x'):
            self.assertEqual(roll('4x2'), 8)
        with self.subTest('*'):
            self.assertEqual(roll('4*2'), 8)
        with self.subTest('/'):
            self.assertEqual(roll('5/2'), 2)
            self.assertEqual(roll('6/2'), 3)

        with self.subTest('**'):
            self.assertEqual(roll('4**2'), 16)

        with self.subTest('+'):
            self.assertEqual(roll('5+6'), 11)
        with self.subTest('-'):
            self.assertEqual(roll('9-2'), 7)

    def test_func(self):
        with self.subTest('min'):
            self.assertEqual(roll('min(3,6)'), 3)
        with self.subTest('max'):
            self.assertEqual(roll('max(3,6)'), 6)

    def test_braces(self):
        with self.subTest('without'):
            self.assertEqual(roll('2+3*4'), 14)
        with self.subTest('with'):
            self.assertEqual(roll('(2+3)*4'), 20)

    def test_junk(self):
        with self.subTest():
            self.assertEqual(roll(' 2 +    3 *\t4'), 14)

    def test_validation(self):
        with self.subTest():
            with self.assertRaises(ArithmeticError):
                roll('print(2)')
        with self.subTest():
            self.assertEqual(roll('log(8, 2)'), 3)


if __name__ == '__main__':
    unittest.main()