def test_msa_exclude__remove_one(): fa_1 = FASTA("A", None, "ACGT") fa_2 = FASTA("B", None, "GCTA") initial = MSA([fa_1, fa_2]) expected = MSA([fa_1]) result = initial.exclude(["B"]) assert_equal(result, expected)
def test_msa_exclude__missing_keys(): msa = MSA([FASTA("Foo", None, "ACGT")]) with pytest.raises(KeyError): msa.exclude(["Bar"])
def test_msa_exclude__no_keys(): msa = MSA([FASTA("Foo", None, "ACGT")]) with pytest.raises(ValueError): msa.exclude([])