def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.low = 0.2
self.chance = 0.2
self.high = 0.21
self.generator_mock = create_autospec(Random)
self.sut = Randomizer(self.generator_mock)
class TestRandomizer(unittest.TestCase):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.low = 0.2
self.chance = 0.2
self.high = 0.21
self.generator_mock = create_autospec(Random)
self.sut = Randomizer(self.generator_mock)
def test_on_high_result_perform_with_chance_should_return_false(self):
self.generator_mock.random.return_value = self.high
assert_that(self.sut.perform_with_chance(self.chance), is_(False))
def test_on_low_result_perform_with_chance_should_return_true(self):
self.generator_mock.random.return_value = self.low
assert_that(self.sut.perform_with_chance(self.chance), is_(True))
def test_should_be_able_to_choose_random_element_from_sequence(self):
# Given
sequence = [1, 2, 3]
self.generator_mock.choice.return_value = 2
# When/Then:
assert_that(self.sut.choice(sequence), is_(equal_to(2)))
def test_should_be_able_to_choose_random_sample_from_sequence(self):
# Given
sequence = [1, 2, 3, 4, 5]
self.generator_mock.sample.return_value = [3, 4, 5]
# When/Then:
assert_that(self.sut.sample(sequence, 3), contains(3, 4, 5))
def test_should_be_able_to_get_uniform_distribution(self):
# Given
self.generator_mock.uniform.return_value = 1.8273
# When/Then:
assert_that(self.sut.uniform(0.5, 2.7))
