from unittest import TestCase from gestform import ( generate_numbers_list, MIN, MAX, randomise_with_iterator, ) class RandomListNumber(TestCase): def setUp(self) -> None: super().setUp() self.generated: list = generate_numbers_list(mini=MIN, maxi=MAX) def test_generate_numbers_list(self): self.assertEqual(min(self.generated), MIN) self.assertEqual(max(self.generated), MAX) self.assertIn(0, self.generated) self.assertNotIn(MIN-1, self.generated) self.assertNotIn(MAX+1, self.generated) def test_randomised_iterator_over_list(self): count: int = 0 for n in randomise_with_iterator(self.generated): with self.subTest(n): self.assertGreaterEqual(n , MIN) self.assertLessEqual(n, MAX) count += 1 self.assertEqual(len(self.generated), count)