-
Notifications
You must be signed in to change notification settings - Fork 59
/
test_vector_visualization.py
44 lines (39 loc) · 1.73 KB
/
test_vector_visualization.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
import unittest
import os
import tempfile
from filecmp import cmp
import py_compile
from pvactools.lib.vector_visualization import VectorVisualization
class FilterTests(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.directory = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
cls.filter_path = os.path.join(cls.directory, "lib", "vector_visualization.py")
cls.test_data_path= os.path.join(cls.directory, "tests", "test_data", "vector_visualization")
cls.spacers = ["AAY", "HHHH", "GGS", "GPGPG", "HHAA", "AAL", "HH", "HHC", "HHH", "HHHD", "HHL", "HHHC"]
def module_compiles(self):
self.assertTrue(py_compile.compile(self.filter_path))
def test_fasta_with_spacers(self):
if 'DISPLAY' in os.environ.keys():
with tempfile.TemporaryDirectory() as output_dir:
self.assertFalse(VectorVisualization(
os.path.join(
self.test_data_path,
'input.spacers.fa'
),
output_dir,
self.spacers
).draw())
self.assertTrue(os.path.exists(os.path.join(output_dir, 'vector.png')))
def test_fasta_with_long_peptide(self):
if 'DISPLAY' in os.environ.keys():
with tempfile.TemporaryDirectory() as output_dir:
self.assertFalse(VectorVisualization(
os.path.join(
self.test_data_path,
'input.long_peptide.fa'
),
output_dir,
self.spacers
).draw())
self.assertTrue(os.path.exists(os.path.join(output_dir, 'vector.png')))