def run(self):
print('Cleaning up stress test files from other runs...')
cleanup(self._cl, StressTestRunner._TAG, should_wait=False)
print('Running stress tests...')
self._start_heartbeat()
self._test_large_bundle()
self.cleanup()
self._test_many_gpu_runs()
self.cleanup()
self._test_multiple_cpus_runs_count()
self.cleanup()
self._test_many_bundle_uploads()
self.cleanup()
self._test_many_worksheet_copies()
self.cleanup()
self._test_parallel_runs()
self.cleanup()
self._test_many_docker_runs()
self.cleanup()
self._test_infinite_memory()
self.cleanup()
self._test_infinite_gpu()
self.cleanup()
self._test_infinite_disk()
self.cleanup()
self._test_many_disk_writes()
self.cleanup()
print('Done.')
def main():
if args.cleanup:
cleanup(cl, SampleWorksheet.TAG)
return
print(args)
ws = SampleWorksheet(cl, args.large, args.preview)
start_time = time.time()
ws.create()
if args.test_print:
ws.test_print()
duration_seconds = time.time() - start_time
print("--- Completion Time: {} minutes---".format(duration_seconds / 60))
def cleanup(self):
if self._args.bypass_cleanup:
return
cleanup(self._cl, StressTestRunner._TAG, not self._args.bypass_wait)