diff --git a/test/test_mmcif.py b/test/test_mmcif.py index 090303fc..31a755fd 100644 --- a/test/test_mmcif.py +++ b/test/test_mmcif.py @@ -1123,6 +1123,8 @@ def test_ensemble_dumper(self): """Test dumping of simple ensembles""" class DummyPostProcess: pass + class DummyModel: + pass m = IMP.Model() s = IMP.pmi.topology.System(m) po = IMP.pmi.mmcif.ProtocolOutput() @@ -1131,10 +1133,14 @@ class DummyPostProcess: pp = DummyPostProcess() pp._id = 99 - po._add_simple_ensemble(pp, 'Ensemble 1', 5, 0.1, 1, - {}, None) - po._add_simple_ensemble(pp, 'Ensemble 2', 5, 0.1, 1, - {}, None) + e = po._add_simple_ensemble(pp, 'Ensemble 1', 5, 0.1, 1, + {}, None) + # Work around python-ihm 1.7 not handling IDs of empty model + # groups properly + e.model_group.append(DummyModel()) + e = po._add_simple_ensemble(pp, 'Ensemble 2', 5, 0.1, 1, + {}, None) + e.model_group.append(DummyModel()) loc = ihm.location.InputFileLocation(repo='foo', path='bar') po.set_ensemble_file(1, loc) loc._id = 42