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