def test_model_description(self):
"""correctly grabs function descriptions"""
all_available = available_models()
for abbrev, desc in all_available.tolist(["Abbreviation", "Description"]):
func = getattr(models_module, abbrev)
doc = func.__doc__.split()
self.assertEqual(desc.split(), doc)
def test_model_by_type(self):
"""correctly obtain models by type"""
for model_type in ["codon", "nucleotide", "protein"]:
table = available_models(model_type)
got = table.distinct_values("Model Type")
self.assertEqual(got, {model_type})
def test_model_abbreviation(self):
"""make sure getting model abbreviations that exist"""
got = set(available_models().tolist("Abbreviation"))
expect = set(["JC69", "CNFGTR", "DSO78"])
self.assertTrue(expect < got)
