class TestUnit::Generators::ScaffoldGenerator

def time?(name)
  attribute = attributes.find { |attr| attr.name == name }
  attribute&.type == :time
end