def test_omission_check_1(): # CC, AA -> AA assert FV.check_omission_trio(np.array([1, 1]), np.array([0, 0]), np.array([0, 0]), 1) assert not FV.check_omission_trio(np.array([1, 1]), np.array([0, 0]), np.array([0, 0]), 0) assert not FV.check_denovo_trio(np.array([1, 1]), np.array([0, 0]), np.array([0, 0]), 0) assert not FV.check_denovo_trio(np.array([1, 1]), np.array([0, 0]), np.array([0, 0]), 1)
def test_weird_denovo_from_ivan_4(): # AC AC CG denovo assert FV.check_denovo_trio(np.array([0, 1]), np.array([0, 1]), np.array([1, 2]), 2) assert not FV.check_denovo_trio(np.array([0, 1]), np.array([0, 1]), np.array([1, 2]), 0) assert not FV.check_denovo_trio(np.array([0, 1]), np.array([0, 1]), np.array([1, 2]), 1) assert not FV.check_omission_trio(np.array([0, 1]), np.array([0, 1]), np.array([1, 2]), 2) assert not FV.check_omission_trio(np.array([0, 1]), np.array([0, 1]), np.array([1, 2]), 0) assert not FV.check_omission_trio(np.array([0, 1]), np.array([0, 1]), np.array([1, 2]), 1)
def test_weird_denovo_from_ivan_1_4(): # AC AA GA denovo assert FV.check_denovo_trio(np.array([0, 1]), np.array([0, 0]), np.array([2, 0]), 2) assert not FV.check_denovo_trio(np.array([0, 1]), np.array([0, 0]), np.array([2, 0]), 0) assert not FV.check_denovo_trio(np.array([0, 1]), np.array([0, 0]), np.array([2, 0]), 1) assert not FV.check_omission_trio(np.array([0, 1]), np.array([0, 0]), np.array([2, 0]), 2) assert not FV.check_omission_trio(np.array([0, 1]), np.array([0, 0]), np.array([2, 0]), 0) assert not FV.check_omission_trio(np.array([0, 1]), np.array([0, 0]), np.array([2, 0]), 1)
def test_omission_not_denovo(): # AA,CC -> AA trio = [ np.array([0, 0]), # p1 np.array([1, 1]), # p2 np.array([0, 0]), # ch ] assert FV.check_omission_trio(*trio, allele_index=1) assert not FV.check_denovo_trio(*trio, allele_index=1) assert not FV.check_mendelian_trio(*trio, allele_index=1) # FIXME: assert not FV.check_omission_trio(*trio, allele_index=0) assert not FV.check_denovo_trio(*trio, allele_index=0) assert FV.check_mendelian_trio(*trio, allele_index=0)
def test_super_weird_denovo_omission_mix_from_ivan_2(): # AA AC TG denovo trio = [ np.array([0, 0]), np.array([0, 1]), np.array([2, 3]), ] assert FV.check_omission_trio(*trio, allele_index=0) assert not FV.check_omission_trio(*trio, allele_index=1) assert not FV.check_omission_trio(*trio, allele_index=2) assert not FV.check_omission_trio(*trio, allele_index=3) assert not FV.check_denovo_trio(*trio, allele_index=0) assert not FV.check_denovo_trio(*trio, allele_index=1) assert FV.check_denovo_trio(*trio, allele_index=2) assert FV.check_denovo_trio(*trio, allele_index=3)
def test_mixed_check_4(): # AA,CC -> AG trio = [ np.array([1, 1]), # p1 np.array([0, 0]), # p2 np.array([0, 2]), # ch ] assert not FV.check_denovo_trio(*trio, allele_index=0) assert not FV.check_denovo_trio(*trio, allele_index=1) assert FV.check_denovo_trio(*trio, allele_index=2) assert not FV.check_omission_trio(*trio, allele_index=0) assert FV.check_omission_trio(*trio, allele_index=1) assert not FV.check_omission_trio(*trio, allele_index=2) assert FV.check_mendelian_trio(*trio, allele_index=0) assert not FV.check_mendelian_trio(*trio, allele_index=1) assert not FV.check_mendelian_trio(*trio, allele_index=2)
def test_mixed_check_1(): # AA,CC -> GG trio = [ np.array([1, 1]), np.array([0, 0]), np.array([2, 2]), ] assert not FV.check_denovo_trio(*trio, allele_index=0) assert not FV.check_denovo_trio(*trio, allele_index=1) assert FV.check_denovo_trio(*trio, allele_index=2) assert FV.check_omission_trio(*trio, allele_index=0) assert FV.check_omission_trio(*trio, allele_index=1) assert not FV.check_omission_trio(*trio, allele_index=2) assert not FV.check_mendelian_trio(*trio, allele_index=0) assert not FV.check_mendelian_trio(*trio, allele_index=1) assert not FV.check_mendelian_trio(*trio, allele_index=2)
def test_omission_check_4(): # AA,CC -> CC assert FV.check_omission_trio(np.array([1, 1]), np.array([0, 0]), np.array([1, 1]), 0) assert not FV.check_omission_trio(np.array([1, 1]), np.array([0, 0]), np.array([1, 1]), 1)
def test_denovo_and_omission_from_ivan_4(): # AA CC CC omission assert FV.check_omission_trio(np.array([0, 0]), np.array([1, 1]), np.array([1, 1]), 0) assert not FV.check_omission_trio(np.array([0, 0]), np.array([1, 1]), np.array([1, 1]), 1)