def test_check_type_default(capsys):
assert not Db.check_type(None)
assert Db.check_type("foo") == "protein"
out, err = capsys.readouterr()
assert err == "Warning: 'foo' is not a valid choice for '_type'. Setting to default 'protein'.\n"
def test_check_type_nuc():
for _input in [
"n", "ncl", "nuc", "dna", "nt", "gene", "transcript", "nucleotide"
]:
assert Db.check_type(_input) == "nucleotide"
def test_check_type_gi():
for _input in ["g", "gi", "gn", "gin", "gi_num", "ginum", "gi_number"]:
assert Db.check_type(_input) == "gi_num"
def test_check_type_protein():
for _input in [
"p", "pr", "prt", "prtn", "prn", "prot", "protn", "protien",
"protein"
]:
assert Db.check_type(_input) == "protein"
