average mean ruby program implementation

maths/average_mean.rb

@@ -0,0 +1,28 @@
+# A ruby program for finding average mean
+
+module AverageMean
+  # Average mean = sum of elements / total number of elements
+  def self.average_mean(n, *array)
+    if n.instance_of? Integer
+      if n == array.size
+        puts "The average mean of the following elements #{array} is #{array.sum/array.size}."
+      else
+        puts "Provide the required #{n} elements properly!"
+      end
+    else
+      raise
+    end
+    rescue
+      puts "Error: Please provide number only!"
+  end
+end
+
+# Valid inputs
+AverageMean.average_mean(2, 3, 1)
+AverageMean.average_mean(5, 1, 2, 3, 4, 5)
+AverageMean.average_mean(3, 2, 2, 2)
+
+# Invalid inputs
+AverageMean.average_mean(2, 3, 1, 5)
+AverageMean.average_mean(2, 3, "a")
+AverageMean.average_mean("a", 1, 2)