def test_drop_keep_estimated_custom_dice_negative(self):
self._test_dice_drop = drop_keep_expression.DropKeepExpression(
"4d[-2,2,100]d2")
self._test_dice_keep = drop_keep_expression.DropKeepExpression(
"4d[-2,2,100]k2")
self.assertEqual(24, self._test_dice_drop.estimated_cost())
self.assertEqual(24, self._test_dice_keep.estimated_cost())
def test_drop_keep_estimated_custom_dice_large_set(self):
self._test_dice_drop = drop_keep_expression.DropKeepExpression(
"4d[-2,0,2,4,6*7,31,-2,-24]d2")
self._test_dice_keep = drop_keep_expression.DropKeepExpression(
"4d[-2,0,2,4,6*7,31,-2,-24]k2")
self.assertEqual(56, self._test_dice_drop.estimated_cost())
self.assertEqual(56, self._test_dice_keep.estimated_cost())
def test_drop_keep_get_probability_distribution_drop_percentile_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("2d%d1")
self.assertEqual(
{i: 1 + 2 * (100 - i)
for i in range(1, 101)},
test_dice.get_probability_distribution().get_result_map(),
)
def test_drop_keep_roll_drop_percentile_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("2d%d1")
roll_set_keep = set()
for _ in range(100000):
roll_set_keep.add(test_dice.roll())
self.assertEqual(1, min(roll_set_keep))
self.assertEqual(100, max(roll_set_keep))
self.assertEqual(100, len(roll_set_keep))
def test_drop_keep_roll_keep_to_zero(self):
test_dice = drop_keep_expression.DropKeepExpression("2d3k0")
roll_set_keep = set()
for _ in range(10000):
roll_set_keep.add(test_dice.roll())
self.assertEqual(0, min(roll_set_keep))
self.assertEqual(0, max(roll_set_keep))
self.assertEqual(1, len(roll_set_keep))
def test_drop_keep_get_probability_distribution_drop_fate_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("3dFd1")
self.assertEqual(
{
-2: 7,
-1: 9,
0: 7,
1: 3,
2: 1
},
test_dice.get_probability_distribution().get_result_map(),
)
def test_drop_keep_get_probability_distribution_keep_custom_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("3d[5,10,15]k2")
self.assertEqual(
{
10: 1,
15: 3,
20: 7,
25: 9,
30: 7
},
test_dice.get_probability_distribution().get_result_map(),
)
def test_drop_keep_get_probability_distribution_drop_to_zero(self):
test_dice = drop_keep_expression.DropKeepExpression("2d3d0")
self.assertEqual(
{
2: 1,
3: 2,
4: 3,
5: 2,
6: 1
},
test_dice.get_probability_distribution().get_result_map(),
)
def test_drop_keep_get_probability_distribution_range(self):
test_dice = drop_keep_expression.DropKeepExpression("3d[-1-1, 0]d1")
self.assertEqual(
{
-2: 10,
-1: 24,
0: 23,
1: 6,
2: 1
},
test_dice.get_probability_distribution().get_result_map(),
)
def test_drop_keep_get_probability_distribution_drop_custom_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("3d[-3,2,4]d1")
self.assertEqual(
{
-6: 7,
-1: 9,
1: 3,
4: 4,
6: 3,
8: 1
},
test_dice.get_probability_distribution().get_result_map(),
)
def test_drop_keep_get_probability_distribution_keep(self):
test_dice = drop_keep_expression.DropKeepExpression("6d4k2")
self.assertEqual(
{
2: 1,
3: 6,
4: 63,
5: 192,
6: 659,
7: 1266,
8: 1909
},
test_dice.get_probability_distribution().get_result_map(),
)
def test_drop_keep_get_probability_distribution_keep_large_number_dice(
self):
test_dice = drop_keep_expression.DropKeepExpression("20d3k4")
self.assertEqual(
{
4: 1,
5: 20,
6: 210,
7: 1520,
8: 1050835,
9: 10485360,
10: 49804890,
11: 149420940,
12: 3276020625,
},
test_dice.get_probability_distribution().get_result_map(),
)
def test_drop_keep_get_probability_distribution_keep_large_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("6d8k2")
self.assertEqual(
{
2: 1,
3: 6,
4: 63,
5: 192,
6: 665,
7: 1458,
8: 3367,
9: 6144,
10: 11523,
11: 18558,
12: 29573,
13: 40512,
14: 52243,
15: 54186,
16: 43653,
},
test_dice.get_probability_distribution().get_result_map(),
)
def test_drop_keep_roll_keep_custom_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("3d[5,10,15]k2")
roll_set_keep = set()
for _ in range(1000):
roll_set_keep.add(test_dice.roll())
self.assertSetEqual({10, 15, 20, 25, 30}, roll_set_keep)
def test_drop_keep_roll_drop_custom_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("3d[-3,2,4]d1")
roll_set_keep = set()
for _ in range(1000):
roll_set_keep.add(test_dice.roll())
self.assertSetEqual({-6, -1, 1, 4, 6, 8}, roll_set_keep)
def test_drop_keep_roll_keep(self):
test_dice = drop_keep_expression.DropKeepExpression("5d10k2")
roll_set_drop = set()
for _ in range(10000):
roll_set_drop.add(test_dice.roll())
self.assertEqual(20, max(roll_set_drop))
def test_drop_keep_roll_drop_fate_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("3dFd1")
roll_set_keep = set()
for _ in range(1000):
roll_set_keep.add(test_dice.roll())
self.assertSetEqual({-2, -1, 0, 1, 2}, roll_set_keep)
def test_drop_keep_get_probability_distribution_drop_to_many(self):
test_dice = drop_keep_expression.DropKeepExpression("2d3d10")
self.assertEqual(
{0: 1},
test_dice.get_probability_distribution().get_result_map())
def test_drop_keep_min_drop_to_zero(self):
test_dice = drop_keep_expression.DropKeepExpression("2d3d0")
self.assertEqual(2, test_dice.min())
def test_drop_keep_str_drop_fate_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("3dFd1")
self.assertEqual("3dFd1", str(test_dice))
def test_drop_keep_estimated_cost_fate_dice(self):
self._test_dice_drop = drop_keep_expression.DropKeepExpression("4dFd2")
self._test_dice_keep = drop_keep_expression.DropKeepExpression("4dFk2")
self.assertEqual(24, self._test_dice_drop.estimated_cost())
self.assertEqual(24, self._test_dice_keep.estimated_cost())
def test_drop_keep_str_drop_custom_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("3d[-3,2,4]d1")
self.assertEqual("3d[-3,2,4]d1", str(test_dice))
def test_drop_keep_str_keep_custom_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("3d[5,10,15]k2")
self.assertEqual("3d[5,10,15]k2", str(test_dice))
def test_drop_keep_estimated_cost_missing_dice_amount(self):
self._test_dice_drop = drop_keep_expression.DropKeepExpression("d6d2")
self._test_dice_keep = drop_keep_expression.DropKeepExpression("d6k2")
self.assertEqual(2, self._test_dice_drop.estimated_cost())
self.assertEqual(6, self._test_dice_keep.estimated_cost())
def test_drop_keep_estimated_cost_percentile_dice(self):
self._test_dice_drop = drop_keep_expression.DropKeepExpression("4d%d2")
self._test_dice_keep = drop_keep_expression.DropKeepExpression("4d%k2")
self.assertEqual(800, self._test_dice_drop.estimated_cost())
self.assertEqual(800, self._test_dice_keep.estimated_cost())
def test_drop_keep_min_drop_to_many(self):
test_dice = drop_keep_expression.DropKeepExpression("2d3d10")
self.assertEqual(0, test_dice.min())
def setUp(self):
self._test_dice_drop = drop_keep_expression.DropKeepExpression("4d6d2")
self._test_dice_keep = drop_keep_expression.DropKeepExpression("4d6k2")
self._mock_parser_gen = mock.create_autospec(rply.ParserGenerator)
def test_drop_keep_str_drop_percentile_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("24d%d12")
self.assertEqual("24d%d12", str(test_dice))
def test_drop_keep_min_drop_custom_dice(self):
test_dice = drop_keep_expression.DropKeepExpression("3d[-3,2,4]d1")
self.assertEqual(-6, test_dice.min())
def test_drop_keep_roll_drop(self):
test_dice = drop_keep_expression.DropKeepExpression("5d10d3")
roll_set_keep = set()
for _ in range(10000):
roll_set_keep.add(test_dice.roll())
self.assertEqual(2, min(roll_set_keep))
