Week2 Assignment prashant kumar choudhary

week_2/pet.rb

@@ -46,10 +46,14 @@ def initialize(name: , animal_type_id: , food_consumed_per_day:)
 
   # Return the habitat of the pet
   def habitat
-    raise NotImplementedError # TODO
+    HABITATS.each do |k,v|
+      if v.include?(animal_type_id)
+          return k
+      end
+    end
   end
 
-  # Returns the cost of food required to feed the animal 
+  # Returns the cost of food required to feed the animal
   # per day
   def food_cost_per_day
     return FOOD_COST_PER_KG[animal_type_id] * food_consumed_per_day
@@ -66,7 +70,7 @@ def food_cost_per_day
   # cat = Pet.new(name: 'cat', animal_type_id: 6, food_consumed_per_day: 0.4)
   # cat.food_required(28) = 11.2 (0.4 * 28)
   def food_required(days)
-    raise NotImplementedError # TODO
+    food_consumed_per_day*days
   end
 
   # This function takes the number of `days` as the input
@@ -76,11 +80,11 @@ def food_required(days)
   # cat = Pet.new(name: 'cat', animal_type_id: 6, food_consumed_per_day: 0.4)
   # cat.food_cost(28) = 8960
   def food_cost(days)
-    raise NotImplementedError # TODO
+    food_cost_per_day * days
   end
 
   # This function takes an array of pets and the `days`
-  # as input and returns the cost to feed them all 
+  # as input and returns the cost to feed them all
   # for the specified number of `days`
   #
   # Eg -
@@ -90,13 +94,13 @@ def food_cost(days)
   # snake = Pet.new(name: 'python', animal_type_id: 4, food_consumed_per_day: 0.3)
   # Pet.cost_to_feed([cat, dog, fish, snake], 6) will return 6180.0
   def self.cost_to_feed(pets, days)
-    raise NotImplementedError # TODO
+    pets.reduce(0) { |tot,pet| tot + pet.food_cost(days) }
   end
 
   # This function takes an array of pets as input
   # and returns a hash with pets name grouped by their animal type
   #
-  # Eg - 
+  # Eg -
   # cat = Pet.new(name: 'cat', animal_type_id: 6, food_consumed_per_day: 0.4)
   # dog = Pet.new(name: 'dog', animal_type_id: 6, food_consumed_per_day: 0.7)
   # fish = Pet.new(name: 'clownfish', animal_type_id: 2, food_consumed_per_day: 0.1)
@@ -110,6 +114,16 @@ def self.cost_to_feed(pets, days)
   #
   # Note - Order is not important
   def self.group_by_animal_type(pets)
-    raise NotImplementedError # TODO
+    temp=pets.group_by { |i| i.animal_type_id }
+
+      pet_group = {}
+
+      temp.each do |id,objs|
+          names = []
+          objs.each { |obj| names<<obj.name }
+          pet_group[id]=names
+      end
+
+      pet_group
   end
 end

week_2/prime_numbers.rb

@@ -1,18 +1,16 @@
-# A prime number is a whole number greater than 1
-# that cannot be exactly divided by any whole number 
-# other than itself and 1 
-# (e.g. 2, 3, 5, 7, 11).
-
-# The function below takes a keyword arguments `n` and 
-# returns an array of prime numbers less than or equal to
-# `n`.
-
-# For example, prime_numbers(n: 20) should return the following:
-# [2, 3, 5, 7, 11, 13, 17, 19]
-
-# If the user gives a invalid input like -4
-# We will raise an `ArgumentError` exception to let the caller know that
-# their function arguments were incorrect.
 def prime_numbers(n:)
-  raise NotImplementedError # TODO
+  raise ArgumentError, "n must be greater than or equal to 2" if n < 2
+
+  primes = []
+  (2..n).each do |i|
+    is_prime = true
+    (2..Math.sqrt(i).ceil).each do |j|
+      if i % j == 0
+        is_prime = false
+        break
+      end
+    end
+    primes << i if is_prime
+  end
+ puts primes
 end