Bug fix: correctly catch loos::StopIteration exceptions for AtomicGroupIterator

tromo · tromo · commit 015bba6949a2 · 2022-10-24T13:59:00.000-04:00
diff --git a/src/AtomicGroup.i b/src/AtomicGroup.i
@@ -71,7 +71,7 @@
 
      pAtom __next__()  {
        if (_idx >= _ag->size())
-	 throw(loos::StopIteration());
+	 throw(StopIteration());
        return((*_ag)[_idx++]);
      }
 
@@ -85,6 +85,7 @@
 
 %}
 
+%catches(loos::StopIteration) loos::AtomicGroupPythonIterator::__next__();
 
 // Translate C++ exception into Python's
 %typemap(throws) loos::StopIteration %{
