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')