class TestUnit::Generators::ScaffoldGenerator

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