def test_e2e_cockroach(): from track.distributed.cockroachdb import CockRoachDB db = CockRoachDB(location='/tmp/cockroach', addrs='localhost:8123') db.start(wait=True) try: end_to_end_train('cockroach://localhost:8123') except Exception as e: raise e finally: db.stop()
def test_e2e_server_socket(): remove('server_test.json') if SKIP_SERVER: return import socket s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.bind(('', 0)) port = s.getsockname()[1] s.close() print('Starting Server') start_track_server('file://server_test.json', 'localhost', port) server = Process(target=start_track_server('file://server_test.json', 'localhost', port)) server.start() print('Starting client') end_to_end_train(f'socket://*****:*****@localhost:{port}') remove('server_test.json')
def test_e2e_cometml(): if SKIP_COMET: return end_to_end_train('cometml:/convnet-test/convnet')
def test_e2e_file(): remove('file_test.json') end_to_end_train('file://file_test.json') remove('file_test.json')
def test_e2e_ephemeral(count=2): end_to_end_train('ephemeral:')
def test_e2e_pickled(count=2): remove('file.pkl') end_to_end_train('pickled://file.pkl') remove('file.pkl')
def test_e2e_mongodb(count=2): end_to_end_train('mongodb://localhost:41033')