Skip to content

Commit

Permalink
Add tests for CoordinateMapping.__eq__
Browse files Browse the repository at this point in the history
  • Loading branch information
Donaim committed Sep 20, 2024
1 parent 41ae45d commit b6e3010
Showing 1 changed file with 24 additions and 1 deletion.
25 changes: 24 additions & 1 deletion tests/test_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,8 @@
from typing import Dict, Union

from aligntools import Cigar, CigarHit, \
connect_cigar_hits, CigarActions, IntDict
connect_cigar_hits, CigarActions, \
IntDict, CoordinateMapping
import aligntools.libexceptions as ex


Expand Down Expand Up @@ -968,3 +969,25 @@ def test_illigal_cigar_hit_to_msa(cigar, reference_seq, query_seq):
obj = parsed_hit(cigar)
with pytest.raises(ex.MSALengthError):
obj.to_msa(reference_seq, query_seq)


def test_coordinate_mapping_eq():
mapping_1 = CoordinateMapping()
mapping_1.extend(2, 3, 0)
mapping_1.extend(3, 4, 1)

mapping_2 = CoordinateMapping()
mapping_2.extend(3, 4, 1)
mapping_2.extend(2, 3, 0)

mapping_3 = CoordinateMapping()
mapping_3.extend(2, 3, 0)
mapping_3.extend(3, 5, 1)

mapping_4 = CoordinateMapping()
mapping_4.extend(2, 3, 1)
mapping_4.extend(3, 4, 2)

assert mapping_1 == mapping_2
assert mapping_1 != mapping_3
assert mapping_1 != mapping_4

0 comments on commit b6e3010

Please sign in to comment.