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)