Movie Problem Solution

movie_problem.rb

@@ -0,0 +1,36 @@
+#!/usr/bin/env ruby
+
+require 'json'
+
+class MovieRating
+    def initialize datafile
+      raw_data = IO.read(datafile)
+      @movies  = JSON.parse(raw_data)
+    rescue Exception => e
+      puts "Caught exception #{e.message} while parsing the json input"
+    end
+
+    def calculate_rating
+      @genres = {}
+      @movies["movies"].each do |mov|
+        mov["genres"].each do |gen|
+            if @genres[gen].nil?
+                @genres[gen] = { :rating => mov["rating"].to_f, :count => 1}
+            else
+                @genres[gen][:rating] += mov["rating"].to_f
+                @genres[gen][:count] += 1
+            end
+        end
+      end
+    end
+
+    def results
+        @genres.each do |name, values|
+            puts "#{(values[:rating]/values[:count]).round(2)} for #{name}"
+        end
+    end
+end
+
+mr = MovieRating.new 'data/movies.json'
+mr.calculate_rating
+mr.results

movie_problem_test.rb

@@ -0,0 +1,30 @@
+require './movie_problem'
+require 'test/unit'
+
+class TestMovieProblem < Test::Unit::TestCase
+    DATA_FILE = "data/movies.json"
+    def test_load_input
+        mp = MovieRating.new(DATA_FILE)
+        assert_not_nil(mp,"Parsing Failed")
+    end
+
+    def test_load_input_no_input
+        mp = MovieRating.new("")
+        assert_not_nil(mp,"Parsing Failed")
+    end
+
+    def test_calculate_rating
+        mp = MovieRating.new(DATA_FILE)
+        assert_nothing_raised(Exception) do
+          mp.calculate_rating
+        end
+    end
+
+    def test_results
+        mp = MovieRating.new(DATA_FILE)
+        assert_nothing_raised(Exception) do
+          mp.calculate_rating
+          mp.results
+        end
+    end
+end