()
| 45 | |
| 46 | |
| 47 | def test_multiprocess_shared_dataloader(): |
| 48 | dataset = DummyDataset(100) |
| 49 | with DataQueue(dataset, producer_num_workers=1) as data_queue: |
| 50 | queue = multiprocessing.Queue() |
| 51 | processes = [] |
| 52 | for _ in range(3): |
| 53 | processes.append(multiprocessing.Process(target=_worker, args=(data_queue, queue))) |
| 54 | processes[-1].start() |
| 55 | for p in processes: |
| 56 | p.join() |
| 57 | assert len(set(_queue_to_list(queue))) == 100 |
| 58 | |
| 59 | |
| 60 | def test_exit_on_crash_finite(): |
no test coverage detected