Refactoring code for Enterprise issue #529

rdt/transformers/text.py

@@ -182,6 +182,9 @@ def _warn_not_enough_unique_values(self, sample_size):
                 f'values (only {max(remaining, 0)} unique values left).'
             )
 
+    def _get_remaining_generator_values(self):
+        return list(self.generator)
+
     def _reverse_transform(self, data):
         """Generate new data using the provided ``regex_format``.
 
@@ -209,9 +212,9 @@ def _reverse_transform(self, data):
             self.generated += sample_size
 
         else:
-            generated_values = list(self.generator)
+            generated_values = self._get_remaining_generator_values()
             reverse_transformed = generated_values[:]
-            self.generated = self.generator_size
+            self.generated += len(generated_values)
             if self.enforce_uniqueness:
                 try:
                     remaining_samples = sample_size - len(reverse_transformed)