def test_without_seed_start(self, mock_ia_seed): augseq = mock.MagicMock() multicore._Pool_initialize_worker(augseq, None) time.sleep(0.01) multicore._Pool_initialize_worker(augseq, None) seed_global_call_1 = mock_ia_seed.call_args_list[0][0][0] seed_local_call_1 = augseq.seed_.call_args_list[0][0][0] seed_global_call_2 = mock_ia_seed.call_args_list[0][0][0] seed_local_call_2 = augseq.seed_.call_args_list[0][0][0] assert (seed_global_call_1 != seed_local_call_1 != seed_global_call_2 != seed_local_call_2), "Got seeds: %d, %d, %d, %d" % ( seed_global_call_1, seed_local_call_1, seed_global_call_2, seed_local_call_2) assert mock_ia_seed.call_count == 2 assert augseq.seed_.call_count == 2
def test_without_seed_start_simulate_py37_or_higher( self, mock_cp, mock_ia_seed, mock_time_ns, mock_vi): def version_info(index): return 3 if index == 0 else 7 mock_vi.__getitem__.side_effect = version_info mock_time_ns.return_value = 1 mock_cp.return_value = mock.MagicMock() mock_cp.return_value.name = "foo" augseq = mock.MagicMock() multicore._Pool_initialize_worker(augseq, None) assert mock_time_ns.call_count == 1 assert mock_ia_seed.call_count == 1 assert augseq.seed_.call_count == 1 seed_global = mock_ia_seed.call_args_list[0][0][0] seed_local = augseq.seed_.call_args_list[0][0][0] assert seed_global != seed_local
def test_with_seed_start(self, mock_ia_pool): augseq = mock.MagicMock() multicore._Pool_initialize_worker(augseq, 1) assert mock_ia_pool._WORKER_SEED_START == 1 assert mock_ia_pool._WORKER_AUGSEQ is augseq assert augseq.localize_random_state_.call_count == 1