def test_main(): """Main function to be called within `__main__` of a test file.""" multi_process_lib.test_main()
cluster = MultiProcessCluster(num_local_workers=0,
num_remote_workers=3)
num_elements = 10
ds = self.make_distributed_range_dataset(num_elements,
cluster,
target_workers="LOCAL")
with self.assertRaisesRegex(errors.InvalidArgumentError,
"no local worker is found"):
get_next = self.getNext(ds)
self.evaluate(get_next())
def testCoordinatedRead(self):
cluster = MultiProcessCluster(num_local_workers=3,
num_remote_workers=3)
ds = dataset_ops.Dataset.range(10).repeat()
ds = self.make_distributed_dataset(ds,
cluster,
job_name="test",
consumer_index=0,
num_consumers=3,
target_workers="LOCAL")
with self.assertRaisesRegex(
errors.InvalidArgumentError,
"Coordinated reads require non-local workers"):
get_next = self.getNext(ds)
self.evaluate(get_next())
if __name__ == "__main__":
multi_process_lib.test_main()
