mirror of
https://github.com/samsledje/D-SCRIPT.git
synced 2026-06-06 16:14:21 +08:00
42 lines
1.8 KiB
Python
42 lines
1.8 KiB
Python
import os
|
|
import shutil
|
|
import subprocess as sp
|
|
|
|
|
|
class TestCommands:
|
|
@classmethod
|
|
def setup_class(cls):
|
|
cmd = "python setup.py install"
|
|
proc = sp.Popen(cmd.split())
|
|
proc.wait()
|
|
os.makedirs("./tmp-dscript-testing/", exist_ok=True)
|
|
|
|
@classmethod
|
|
def teardown_class(cls):
|
|
shutil.rmtree("./tmp-dscript-testing/")
|
|
|
|
def _run_command(self, cmd):
|
|
proc = sp.Popen(cmd.split())
|
|
proc.wait()
|
|
assert not proc.returncode
|
|
|
|
def test_embed(self):
|
|
cmd = "dscript embed --seqs dscript/tests/test.fasta --outfile tmp-dscript-testing/test_embed.h5"
|
|
self._run_command(cmd)
|
|
|
|
def test_train_with_topsy_turvy(self):
|
|
cmd = "dscript train --topsy-turvy --train dscript/tests/test.csv --test dscript/tests/test.csv --embedding tmp-dscript-testing/test_embed.h5 --outfile tmp-dscript-testing/test_tt-train.log --save-prefix tmp-dscript-testing/test_train"
|
|
self._run_command(cmd)
|
|
|
|
def test_train_without_topsy_turvy(self):
|
|
cmd = "dscript train --train dscript/tests/test.csv --test dscript/tests/test.csv --embedding tmp-dscript-testing/test_embed.h5 --outfile tmp-dscript-testing/test_tt-train.log --save-prefix tmp-dscript-testing/test_train"
|
|
self._run_command(cmd)
|
|
|
|
def test_evaluate(self):
|
|
cmd = "dscript evaluate --test dscript/tests/test.csv --embedding tmp-dscript-testing/test_embed.h5 --model tmp-dscript-testing/test_train_final.sav --outfile tmp-dscript-testing/test_evaluate"
|
|
self._run_command(cmd)
|
|
|
|
def test_predict(self):
|
|
cmd = "dscript predict --seqs dscript/tests/test.fasta --pairs dscript/tests/test.csv --model tmp-dscript-testing/test_train_final.sav --outfile tmp-dscript-testing/test_predict --thresh 0.05"
|
|
self._run_command(cmd)
|