xdice/test.py

'''
Created on 18 nov. 2016

@author: olinox
'''
import unittest

import xdice


class Test(unittest.TestCase):
    """unitests for DiceRollParser"""

    def test_secured_eval(self):
        # refused
        self.assertRaises(TypeError, xdice._secured_eval, "open('foo', 'r')")
        # accepted
        xdice._secured_eval("1 + max([1,2,3])")
        xdice._secured_eval("1 + min([1,2,3])")

    def test_patterns_validation(self):
        """check the with behaviour with valid / invalid patterns"""
        # check valid expressions
        xdice.roll("3")
        xdice.roll("0d6")
        xdice.roll("1d6")
        xdice.roll("1d66")
        xdice.roll("1d4+2")
        xdice.roll("1d4-2")
        xdice.roll("1d4 - 2 ")
        xdice.roll("6d102+123")
        xdice.roll("6d102+123+8d6")
        xdice.roll("6d102+123+8d6+2+1+1-8-6d4+8-2d101+100d2")
        xdice.roll("32+3-0-2")
        xdice.roll("1d1-3")
        xdice.roll("1000d1000")
        xdice.roll("10 d 10")
        xdice.roll("10*1d10/2")
        xdice.roll("1d20**2")
        xdice.roll("abs(1d6-1d10)")
        xdice.roll("max(1d6,2d4)")
        xdice.roll("min(1d6,2d4)")
        xdice.roll("d")
        xdice.roll("2d")
        xdice.roll("d6")
        xdice.roll("3d6l")
        xdice.roll("3d6l2")
        xdice.roll("3d6h")
        xdice.roll("3d6h2")
        xdice.roll("6d6lh")
        xdice.roll("6d6lh2")
        xdice.roll("6d6l2h")
        xdice.roll("6d6l2h2")
        xdice.roll("3dlh")
        xdice.roll("1d%")
        xdice.roll("d%")
        xdice.roll("1+R3(1d6+1)")
        xdice.roll("3d6!")
        xdice.roll("3d6x")
        xdice.roll("3d6h1x")

        # test invalid expressions
        self.assertRaises(ValueError, xdice.roll, "")
        self.assertRaises(ValueError, xdice.roll, "1d0")
        self.assertRaises(TypeError, xdice.roll, "abc")
        self.assertRaises(TypeError, xdice.roll, "1d2,3")
        self.assertRaises(ValueError, xdice.roll, "1d6l2")
        self.assertRaises(ValueError, xdice.roll, "1d6h2")
        self.assertRaises(ValueError, xdice.roll, "1d6lh")
        self.assertRaises(SyntaxError, xdice.roll, "1+R3(1d6+1")

    def test_dice_object(self):

        d = xdice.Dice(6, 6)
        self.assertEqual(d.sides, 6)
        self.assertEqual(d.amount, 6)
        self.assertEqual(d.__repr__(), "<Dice; sides=6; amount=6>")
        d.drop_lowest = 1
        d.drop_highest = 1
        self.assertEqual(d.__repr__(), "<Dice; sides=6; amount=6; drop_lowest=1; drop_highest=1>")
        d.explode = True
        self.assertEqual(d.__repr__(), "<Dice; sides=6; amount=6; drop_lowest=1; drop_highest=1; explode>")

        self.assertRaises(ValueError, setattr, d, "sides", -1)
        self.assertRaises(ValueError, setattr, d, "sides", "a")
        self.assertRaises(ValueError, setattr, d, "sides", None)
        self.assertRaises(ValueError, setattr, d, "amount", -1)
        self.assertRaises(ValueError, setattr, d, "amount", "a")
        self.assertRaises(ValueError, setattr, d, "amount", None)
        self.assertRaises(ValueError, setattr, d, "drop_lowest", -1)
        self.assertRaises(ValueError, setattr, d, "drop_lowest", "a")
        self.assertRaises(ValueError, setattr, d, "drop_lowest", None)
        self.assertRaises(ValueError, setattr, d, "drop_highest", -1)
        self.assertRaises(ValueError, setattr, d, "drop_highest", "a")
        self.assertRaises(ValueError, setattr, d, "drop_highest", None)

        self.assertEqual(xdice.Dice(1, 6).roll(), 6)
        self.assertEqual(xdice.Dice(1, 6, 1).roll(), 5)
        self.assertEqual(xdice.Dice(1, 6, 0, 1).roll(), 5)
        self.assertEqual(xdice.Dice(1, 6, 1, 1).roll(), 4)
        self.assertEqual(xdice.Dice(1, 6, 3, 3).roll(), 0)

        self.assertEqual(xdice.Dice.parse("6d1").roll(), 6)
        self.assertRaises(ValueError, xdice.Dice.parse, "a1d6")
        self.assertEqual(xdice.Dice.parse("6d1h1").roll().name, "6d1h1")
        self.assertEqual(xdice.Dice.parse("6 D 1h1").roll().name, "6d1h1")

        self.assertEqual(xdice.Dice.parse("3d1!").roll(), 6)
        self.assertEqual(xdice.Dice.parse("3d1x").roll(), 6)
        self.assertEqual(xdice.Dice.parse("3d1lhx").roll(), 2)

    def test_score_object(self):

        s = xdice.Score([1, 2, 3])

        self.assertEqual(s, 6)
        self.assertEqual(str(s), "6")
        self.assertEqual(list(s), [1, 2, 3])
        self.assertEqual(s.detail, [1, 2, 3])
        self.assertTrue(1 in s)
        self.assertEqual(s.__repr__(), "<Score; score=6; detail=[1, 2, 3]; dropped=[]; name=>")

        s = xdice.Score([1, 2, 3], dropped=[1], name='foo')
        self.assertEqual(s.__repr__(), "<Score; score=6; detail=[1, 2, 3]; dropped=[1]; name=foo>")

    def test_pattern_object(self):

        self.assertEqual(xdice._normalize("1 D 6"), "1d6")

        p = xdice.Pattern("6d1+6")
        p.compile()
        self.assertEqual(p.format_string, "{0}+6")
        self.assertEqual(p.dices, [xdice.Dice(1, 6)])
        self.assertEqual(p.roll(), 12)

        p = xdice.Pattern("R2(6d1+6)")
        p.compile()
        self.assertEqual(p.format_string, "({0}+6+{1}+6)")
        self.assertEqual(p.dices, [xdice.Dice(1, 6), xdice.Dice(1, 6)])
        self.assertEqual(p.roll(), 24)

    def test_patternscore_objet(self):
        ps = xdice.PatternScore("{0}+6", [xdice.Score([1, 1, 1, 1, 1, 1])])

        self.assertEqual(ps, 12)
        self.assertEqual(ps.score(0), 6)
        self.assertEqual(ps.scores(), [6])
        self.assertEqual(ps.format(), "[1, 1, 1, 1, 1, 1]+6")
        self.assertEqual(ps.format(verbose=True), "(scores:[1, 1, 1, 1, 1, 1])+6")

    def test_compile(self):
        p1 = xdice.compile("6d1+6")
        p2 = xdice.Pattern("6d1+6")
        p2.compile()

        self.assertEqual(p1.format_string, p2.format_string)
        self.assertEqual(p1.dices, p2.dices)

    def test_roll(self):
        ps1 = xdice.roll("6d1+6")
        ps2 = xdice.PatternScore("{0}+6", [xdice.Score([1, 1, 1, 1, 1, 1])])

        self.assertEqual(ps1, ps2)

    def test_rolldice(self):
        self.assertEqual(xdice.rolldice(1, 6), xdice.Dice(1, 6).roll())

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