-
Notifications
You must be signed in to change notification settings - Fork 1
/
test_net_chop.py
58 lines (55 loc) · 1.77 KB
/
test_net_chop.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import unittest
import unittest.mock
import os
import re
import sys
import tempfile
import py_compile
from filecmp import cmp
pvac_dir = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
sys.path.append(pvac_dir)
import pvacseq.lib
import pvacseq
def make_response(data, files, path):
reader = open(os.path.join(
path,
'net_chop_%s.html'%data['method']
), mode='rb')
response_obj = lambda :None
response_obj.status_code = 200
response_obj.content = reader.read()
reader.close()
return response_obj
class NetChopTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.script_path = os.path.dirname(pvacseq.__file__) + '/lib/' + 'net_chop.py'
cls.test_data_directory = os.path.join(
pvac_dir,
'tests',
'test_data',
'net_chop'
)
cls.methods=['cterm', '20s']
cls.request_mock = unittest.mock.Mock(side_effect = lambda url, data, files=None: make_response(
data,
files,
cls.test_data_directory
))
pvacseq.lib.net_chop.requests.post = cls.request_mock
def test_net_chop_compiles(self):
compiled_script_path = py_compile.compile(self.script_path)
self.assertTrue(compiled_script_path)
def test_net_chop_runs(self):
for method in self.methods:
output_file = tempfile.NamedTemporaryFile()
pvacseq.lib.net_chop.main([
os.path.join(self.test_data_directory, 'Test_filtered.tsv'),
output_file.name,
'--method',
method
])
self.assertTrue(cmp(
os.path.join(self.test_data_directory, 'output_%s.tsv'%method),
output_file.name
))