Use correct object

schrodinger · Aug 7, 2024 · ab151df · ab151df
1 parent 58b5709
commit ab151df
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/layer3/MoleculeExporter.cpp b/layer3/MoleculeExporter.cpp
@@ -1820,7 +1820,7 @@ struct MoleculeExporterChemPy : public MoleculeExporter {
 
   void writeAtom() override {
     PyObject *atom = CoordSetAtomToChemPyAtom(G,
-        m_iter.getAtomInfo(), m_last_obj, m_coord, getRefPtr(),
+        m_iter.getAtomInfo(), m_iter.obj, m_coord, getRefPtr(),
         m_iter.getAtm(), m_mat_full.ptr);
 
     if (atom) {

diff --git a/testing/tests/api/exporting.py b/testing/tests/api/exporting.py
@@ -353,8 +353,11 @@ def testGetModelObjectName(self):
         cmd.load(self.datafile('1rna.cif'))
 
         m1 = cmd.get_model()
+        cnt = cmd.count_atoms('%1oky-frag')
+
         self.assertEqual(m1.atom[0].model, '1oky-frag')
         self.assertEqual(m1.atom[-1].model, '1rna')
+        self.assertEqual(m1.atom[cnt].model, '1rna')
 
 
     @testing.requires_version('2.1')