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_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_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_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_validation(self):
with self.subTest():
with self.assertRaises(ArithmeticError):
roll('print(2)')
with self.subTest():
self.assertEqual(roll('log(8, 2)'), 3)
def test_junk(self):
with self.subTest():
self.assertEqual(roll(' 2 + 3 *\t4'), 14)
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_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_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))