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