def test_will_correct_model_only(self):
unit = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id=None,
symmetry='1_555')
assert self.loader.correct_model_only(unit) == unit._replace(model=1)
def test_knows_not_to_alter_viral_and_alt(self):
unit = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id=None,
symmetry='6_555')
alt = unit._replace(symmetry='1_555', alt_id='B')
assert self.loader.correct_viral_and_alt(unit) is None
assert self.loader.correct_viral_and_alt(alt) is None
def test_will_correct_alt_id_and_model_if_needed(self):
unit = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id=None,
symmetry='6_765')
ans = Unit(pdb='1Z7F',
model=1,
chain='A',
component_number=6,
insertion_code=None,
alt_id='A',
symmetry='6_765')
assert self.loader.correct_model_and_alt_id(unit) == ans
def test_can_alter_to_viral_with_alt_id(self):
unit = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id=None,
symmetry='1_555')
ans = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id='A',
symmetry='P_1')
assert self.loader.correct_viral_and_alt(unit) == ans
def test_can_correct_to_viral_operator(self):
unit = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id=None,
symmetry='1_555')
ans = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id=None,
symmetry='P_1')
assert self.loader.correct_to_viral_p_1_operator(unit) == ans
def will_not_correct_alt_id_if_not_needed(self):
unit = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id='B',
symmetry='6_765')
assert self.loader.correct_alt_id(unit) is None
def test_will_set_model_to_one_if_not_needed(self):
unit = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id=None,
symmetry='1_555')
assert self.loader.correct_model(unit) is None
def test_can_turn_a_unit_id_to_unit(self):
unit = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id=None,
symmetry='6_765')
assert self.loader.as_unit('1Z7F|2|A|C|6||||6_765') == unit
def test_will_not_alter_if_non_standard_operator(self):
unit = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id=None,
symmetry='6_765')
assert self.loader.correct_to_viral_p_1_operator(unit) is None
def test_can_do_nothing_to_unit(self):
unit = Unit(pdb='1Z7F',
model=2,
chain='A',
component_number=6,
insertion_code=None,
alt_id=None,
symmetry='6_765')
assert self.loader.correct_nothing(unit) == unit