def test_make_vae_dataset():
assert not bash("cd ~/Code/navrep/navrep/scripts && \
ipython -- make_vae_dataset.py --environment navreptrain --n 1")
for i in range(1, 101):
assert not bash(
"cp ~/navrep/datasets/V/navreptrain/000_scans_robotstates_actions_rewards_dones.npz \
~/navrep/datasets/V/navreptrain/{:03}_scans_robotstates_actions_rewards_dones.npz"
.format(i))
def test_navrep_is_clean():
if os.path.isdir(os.path.expanduser("~/navrep/models")) or \
os.path.isdir(os.path.expanduser("~/navrep/logs")) or \
os.path.isdir(os.path.expanduser("~/navrep/eval")):
bash("echo $(find ~/navrep -type f | wc -l) files found in ~/navrep!")
if 'OVERWRITE_NAVREP' in os.environ and os.environ[
'OVERWRITE_NAVREP'] == '1':
print(
"WARNING: OVERWRITE_NAVREP is set to 1, will overwrite files in ~/navrep dir"
)
else:
print(
"""ERROR: running tests will overwrite data in the ~/navrep folder!
if you really wish to continue, run OVERWRITE_NAVREP=1 pytest"""
)
assert False
def test_gym_vae_lstm_v_only():
assert not bash("cd ~/Code/navrep/navrep/scripts && \
ipython -- train_gym_navreptrainencodedenv.py --backend VAE_LSTM --encoding V_ONLY --no-gpu --n 100000"
)
def test_train_rnn():
assert not bash("cd ~/Code/navrep/navrep/scripts && \
ipython -- train_rnn.py --environment navreptrain --n 60")
def test_make_rnn_dataset():
assert not bash("cd ~/Code/navrep/navrep/scripts && \
ipython -- make_rnn_dataset.py --environment navreptrain --n 3")
def test_train_vae():
assert not bash("cd ~/Code/navrep/navrep/scripts && \
ipython -- train_vae.py --environment navreptrain --n 1000")
def test_gym_gpt_vm():
assert not bash("cd ~/Code/navrep/navrep/scripts && \
ipython -- train_gym_navreptrainencodedenv.py --backend GPT --encoding VM --n 100000"
)
def test_gym_gpt1d_m_only():
assert not bash("cd ~/Code/navrep/navrep/scripts && \
ipython -- train_gym_navreptrainencodedenv.py --backend GPT1D --encoding M_ONLY --n 100000"
)
def test_gym_vae1dlstm_vm():
assert not bash("cd ~/Code/navrep/navrep/scripts && \
ipython -- train_gym_navreptrainencodedenv.py --backend VAE1DLSTM --encoding VM --n 100000"
)
