def test_assembler(): stmt = make_stmt() ha = HtmlAssembler([stmt]) result = ha.make_model() assert isinstance(result, str) # Read from the template file and make sure the beginning and end of the # content matches template, _, _ = loader.get_source(None, 'indra/template.html') assert result.startswith(template[0:100]) # Make sure assembler works with other parameters provided stmt2 = make_bad_stmt() ha = HtmlAssembler(source_counts={ stmt.get_hash(): { 'test': 1 }, stmt2.get_hash(): { 'test': 1 } }, ev_counts={ stmt.get_hash(): 1, stmt2.get_hash(): 1 }, db_rest_url='test.db.url') ha.add_statements([stmt, stmt2]) result = ha.make_model(grouping_level='agent-pair') assert isinstance(result, str) result = ha.make_model(grouping_level='statement') assert isinstance(result, str) # Check simple=False result = ha.make_model(grouping_level='statement', simple=False) assert isinstance(result, str) # Test belief badges result = ha.make_model(grouping_level='statement', show_belief=True) assert isinstance(result, str) assert '<small\n' \ ' class="badge badge-pill badge-belief"\n' \ ' title="Belief score for this statement">1.0</small>' in result result = ha.make_model(grouping_level='statement', show_belief=False) assert isinstance(result, str) assert '<small\n' \ ' class="badge badge-pill badge-belief"\n' \ ' title="Belief score for this statement">1</small>' \ not in result # Test if source URL exists assert 'http://www.causalbionet.com/' in result # Make sure warning can be appended ha.append_warning('warning') assert ('\t<span style="color:red;">(CAUTION: warning occurred when ' 'creating this page.)</span>' in ha.model) # Make sure model is created before saving ha = HtmlAssembler([stmt]) assert not ha.model ha.save_model('tempfile.html') assert ha.model
def test_assembler(): stmt = make_stmt() ha = HtmlAssembler([stmt]) result = ha.make_model() assert isinstance(result, str) # Read from the template file and make sure the beginning and end of the # content matches template, _, _ = loader.get_source(None, 'indra/template.html') assert result.startswith(template[0:100]) # Make sure assembler works with other parameters provided stmt2 = make_bad_stmt() ha = HtmlAssembler(source_counts={ stmt.get_hash(): { 'test': 1 }, stmt2.get_hash(): { 'test': 1 } }, ev_totals={ stmt.get_hash(): 1, stmt2.get_hash(): 1 }, db_rest_url='test.db.url') ha.add_statements([stmt, stmt2]) result = ha.make_model(with_grouping=True) assert isinstance(result, str) result = ha.make_model(with_grouping=False) assert isinstance(result, str) # Make sure warning can be appended ha.append_warning('warning') assert ('\t<span style="color:red;">(CAUTION: warning occurred when ' 'creating this page.)</span>' in ha.model) # Make sure model is created before saving ha = HtmlAssembler([stmt]) assert not ha.model ha.save_model('tempfile.html') assert ha.model