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
