import os
from indra.assemblers.tsv_assembler import TsvAssembler
from indra.sources.signor import SignorProcessor

# Get some statements from Signor
sp = SignorProcessor()
stmts = sp.statements

def test_tsv_init():
    ta = TsvAssembler(stmts)
    ta.make_model('tsv_test')

def test_tsv_add_stmts():
    ta = TsvAssembler()
    ta.add_statements(stmts)
    assert len(ta.statements) == len(stmts)

def test_make_model():
    ta = TsvAssembler(stmts)
    ta.make_model('tsv_test.tsv')
    assert os.path.exists('tsv_test.tsv')



@@ -3,16 +3,49 @@
 port os%0A
+from indra.sources import signor%0A
 from ind
@@ -96,113 +96,169 @@
 ler%0A
-from indra.sources.signor import SignorProcessor%0A%0A# Get some statements from Signor
+%0A# Get some statements from Signor%0Afrom .test_signor import test_data_file, test_complexes_file
 %0Asp = 
-S
+s
 ignor
-P
+.p
 rocess
-or(
+_from_file(test_data_file, test_complexes_file
 )%0Ast

Fix TSV Assembler reference to Signor files