()
| 533 | |
| 534 | |
| 535 | def test_uniform(): |
| 536 | def pipe(max_batch_size, input_data, device): |
| 537 | pipe = Pipeline(batch_size=max_batch_size, num_threads=4, device_id=0) |
| 538 | dist = fn.random.uniform() |
| 539 | data = fn.external_source(source=input_data, cycle=False, device=device) |
| 540 | processed = data * dist |
| 541 | pipe.set_outputs(processed) |
| 542 | return pipe |
| 543 | |
| 544 | run_pipeline(generate_data(31, 13, array_1d_shape_generator), pipeline_fn=pipe) |
| 545 | |
| 546 | |
| 547 | def test_random_choice(): |
nothing calls this directly
no test coverage detected