lib/minitest/spec.rb in minitest-5.4.0 vs lib/minitest/spec.rb in minitest-5.4.1

@@ -204,13 +204,15 @@
       @specs ||= 0
       @specs += 1
 
       name = "test_%04d_%s" % [ @specs, desc ]
 
+      undef_klasses = self.children.reject { |c| c.public_method_defined? name }
+
       define_method name, &block
 
-      self.children.each do |mod|
-        mod.send :undef_method, name if mod.public_method_defined? name
+      undef_klasses.each do |undef_klass|
+        undef_klass.send :undef_method, name
       end
 
       name
     end