def setUp(self):
self._probability_distribution_factory = ProbabilityDistributionFactory()
self._mock_syntax = [create_autospec(IDiceExpression) for _ in range(2)]
self._mock_syntax[0].roll.return_value = 2
self._mock_syntax[0].max.return_value = 8
self._mock_syntax[0].min.return_value = 6
self._mock_syntax[0].__str__.return_value = "7"
self._mock_syntax[0].estimated_cost.return_value = 9
self._mock_syntax[0].get_probability_distribution.return_value = self._probability_distribution_factory.create(
{-3: 1, 0: 2, 4: 1}
)
self._mock_syntax[0].get_contained_variables.return_value = {"mock one"}
self._mock_syntax[1].roll.return_value = 4
self._mock_syntax[1].max.return_value = 6
self._mock_syntax[1].min.return_value = 8
self._mock_syntax[1].__str__.return_value = "2d8"
self._mock_syntax[1].estimated_cost.return_value = 7
self._mock_syntax[1].get_probability_distribution.return_value = self._probability_distribution_factory.create(
{-3: 2, 0: 1, 1: 2, 4: 1}
)
self._mock_syntax[1].get_contained_variables.return_value = {"mock two"}
self._test_binary_operator = BinaryOperatorExpression("<", self._mock_syntax[0], self._mock_syntax[1])
self._mock_parser_gen = create_autospec(rply.ParserGenerator)
def test_binary_operator_add_production_function(self):
BinaryOperatorExpression.add_production_function(
self._mock_parser_gen, self._probability_distribution_factory)
self._mock_parser_gen.production.assert_called_once_with(
"""expression : expression BINARY_OPERATOR expression""")
class TestBinaryOperatorGreaterThanExpression(TestCase):
def setUp(self):
self._probability_distribution_factory = ProbabilityDistributionFactory(
)
self._mock_syntax = [
create_autospec(IDiceExpression) for _ in range(2)
]
self._mock_syntax[0].roll.return_value = 9
self._mock_syntax[0].max.return_value = 8
self._mock_syntax[0].min.return_value = 6
self._mock_syntax[0].__str__.return_value = "7"
self._mock_syntax[0].estimated_cost.return_value = 9
self._mock_syntax[
0].get_probability_distribution.return_value = self._probability_distribution_factory.create(
{
-3: 1,
0: 2,
4: 1
})
self._mock_syntax[0].get_contained_variables.return_value = {
"mock one"
}
self._mock_syntax[1].roll.return_value = 4
self._mock_syntax[1].max.return_value = 6
self._mock_syntax[1].min.return_value = 8
self._mock_syntax[1].__str__.return_value = "2d8"
self._mock_syntax[1].estimated_cost.return_value = 7
self._mock_syntax[
1].get_probability_distribution.return_value = self._probability_distribution_factory.create(
{
-3: 2,
0: 1,
1: 2,
4: 1
})
self._mock_syntax[1].get_contained_variables.return_value = {
"mock two"
}
self._test_binary_operator = BinaryOperatorExpression(
">", self._mock_syntax[0], self._mock_syntax[1])
self._mock_parser_gen = create_autospec(rply.ParserGenerator)
def test_binary_operator_add_production_function(self):
BinaryOperatorExpression.add_production_function(
self._mock_parser_gen, self._probability_distribution_factory)
self._mock_parser_gen.production.assert_called_once_with(
"""expression : expression BINARY_OPERATOR expression""")
def test_gt_roll_true(self):
for _ in range(100):
self.assertEqual(1, self._test_binary_operator.roll())
def test_gt_roll_false(self):
self._mock_syntax[0].roll.return_value = 2
for _ in range(100):
self.assertEqual(0, self._test_binary_operator.roll())
def test_gt_max_true(self):
self.assertEqual(1, self._test_binary_operator.max())
def test_gt_max_false(self):
self._mock_syntax[
0].get_probability_distribution.return_value = self._probability_distribution_factory.create(
{5: 1})
self._mock_syntax[
1].get_probability_distribution.return_value = self._probability_distribution_factory.create(
{5: 1})
self.assertEqual(0, self._test_binary_operator.max())
def test_gt_min_true(self):
self._mock_syntax[
0].get_probability_distribution.return_value = self._probability_distribution_factory.create(
{
5: 1,
6: 1
})
self._mock_syntax[
1].get_probability_distribution.return_value = self._probability_distribution_factory.create(
{4: 1})
self.assertEqual(1, self._test_binary_operator.min())
def test_gt_min_false(self):
self.assertEqual(0, self._test_binary_operator.min())
def test_gt_str(self):
self.assertEqual("7 > 2d8", str(self._test_binary_operator))
def test_gt_estimated_cost(self):
self.assertEqual(16, self._test_binary_operator.estimated_cost())
def test_gt_get_probability_distribution(self):
self.assertEqual(
{
0: 15,
1: 9
},
self._test_binary_operator.get_probability_distribution().
get_result_map(),
)
def test_gt_get_contained_variables(self):
self.assertSetEqual(
{"mock one", "mock two"},
self._test_binary_operator.get_contained_variables(),
)