def test_all_inputs():
import os
import json
from EvoDAG.command_line import CommandLineParams
fname = training_set()
sys.argv = [
'EvoDAG', '-C', '--parameters', 'cache.evodag', '-p3', '-e2', '-r',
'2', fname
]
c = CommandLineParams()
c.parse_args()
with open('cache.evodag') as fpt:
a = json.loads(fpt.read())[0]
assert 'all_inputs' in a
os.unlink('cache.evodag')
default_nargs()
def test_word2id2():
import tempfile
import os
from EvoDAG.command_line import CommandLineParams
fname = tempfile.mktemp()
id2word = dict([[0, 'a'], [1, 'b'], [2, 'c']])
with open(fname, 'w') as fpt:
for x, v in zip(X, cl):
l = x.tolist()
l.append(id2word[v])
fpt.write(','.join(map(str, l)))
fpt.write('\n')
sys.argv = [
'EvoDAG', '-C', '--parameters', 'cache.evodag', '-p3', '-e1', '-r',
'2', fname
]
c = CommandLineParams()
c.parse_args()
print(len(c.word2id))
assert len(c.word2id) == 0
os.unlink('cache.evodag')
os.unlink(fname)
default_nargs()
def test_nan():
import numpy as np
from EvoDAG.command_line import CommandLineParams
c = CommandLineParams()
assert np.isfinite(c.inputs.convert_label('NaN'))
default_nargs()
