def test_setup_shuffler_shuffle_none(self): runner = DiscoverRunner(shuffle=None) self.assertIsNone(runner.shuffle) with mock.patch("random.randint", return_value=1): with captured_stdout() as stdout: runner.setup_shuffler() self.assertEqual(stdout.getvalue(), "Using shuffle seed: 1 (generated)\n") self.assertEqual(runner.shuffle_seed, 1)
def test_setup_shuffler_shuffle_int(self): runner = DiscoverRunner(shuffle=2) self.assertEqual(runner.shuffle, 2) with captured_stdout() as stdout: runner.setup_shuffler() expected_out = "Using shuffle seed: 2 (given)\n" self.assertEqual(stdout.getvalue(), expected_out) self.assertEqual(runner.shuffle_seed, 2)
def test_setup_shuffler_no_shuffle_argument(self): runner = DiscoverRunner() self.assertIs(runner.shuffle, False) runner.setup_shuffler() self.assertIsNone(runner.shuffle_seed)