diff --git a/student.rb b/student.rb index 514cc09..17a72a4 100644 --- a/student.rb +++ b/student.rb @@ -7,6 +7,10 @@ def initialize(first_name, last_name, grade) @grade = grade end + def junior? + grade ==11 + end + def senior? grade == 12 end @@ -16,15 +20,25 @@ def to_s end end +def juniors(students) + students.select{|student| student.junior?} +end + def seniors(students) students.select { |student| student.senior? } end + fred = Student.new("Fred", "James", 12) sarah = Student.new("Sarah", "Smith", 12) jack = Student.new("Jack", "Gong", 11) all_students = [fred, sarah, jack] -seniors(all_students).each do |student| - puts student -end +puts "Seniors" +seniors(all_students).select{|student| puts "*#{student}"} +puts +puts "Juniors" +juniors(all_students).select{|student| puts "*#{student}"} + + +