From edfed6c04eb3b5734bb74b9048f67f9e0cc45a01 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Martin=20Ho=C5=99e=C5=88ovsk=C3=BD?=
 <martin.horenovsky@gmail.com>
Date: Sun, 15 Feb 2026 21:43:54 +0100
Subject: [PATCH] Efficient skipToNthElementImpl for Take and Map generators

---
 .../generators/catch_generators_adapters.hpp  | 22 ++++---------------
 1 file changed, 4 insertions(+), 18 deletions(-)

diff --git a/src/catch2/generators/catch_generators_adapters.hpp b/src/catch2/generators/catch_generators_adapters.hpp
index c46cb927..7e21671a 100644
--- a/src/catch2/generators/catch_generators_adapters.hpp
+++ b/src/catch2/generators/catch_generators_adapters.hpp
@@ -30,13 +30,8 @@ namespace Generators {
                     "Coud not jump to Nth element: not enough elements" );
             }
 
-            for (; m_returned < n; ++m_returned) {
-                const auto success = m_generator.next();
-                if ( !success ) {
-                    Detail::throw_generator_exception(
-                        "Coud not jump to Nth element: not enough elements" );
-                }
-            }
+            m_generator.skipToNthElement( n );
+            m_returned = n;
         }
 
     public:
@@ -186,17 +181,8 @@ namespace Generators {
         mutable Optional<T> m_cache;
 
         void skipToNthElementImpl( std::size_t n ) override {
-            for ( size_t curr = GeneratorUntypedBase::currentElementIndex();
-                  curr < n;
-                  ++curr ) {
-                const auto success = m_generator.next();
-                if (!success) {
-                    Detail::throw_generator_exception(
-                        "Coud not jump to Nth element: not enough elements" );
-                }
-            }
-
-            m_cache = m_function( m_generator.get() );
+            m_generator.skipToNthElement( n );
+            m_cache.reset();
         }
 
     public: