diff --git a/GemCity.rb b/GemCity.rb deleted file mode 100644 index 690dc84..0000000 --- a/GemCity.rb +++ /dev/null @@ -1,56 +0,0 @@ -class GemCity - -=begin -This class represents the town of GemCity -This is a town riddled with crime but we can find out how happy the town is -=end - def initialize - @people, @population = { - :thieves => 5, - :Officers => 1}, - 50 - end - - def thieves thieves_number=@people[:thieves] - return @people[:thieves] = thieves_number - end - - def officers - return @people[:Officers] - end - - def population; return @population; end - - def setOfficers officers - @people[:Officers] = officers - end - - def happiness_of_town - #happiness is random... people don't know what they want! - happinessVals = Array.new - happiness = 0 - for index in (1..@population) - happinessVals.push(rand((100 - successful_crime_rate) .. 100)) - index += 1 - end - happinessVals.each do |value| - happiness += value - end - return happiness / 100 - end - - def successful_crime_rate - thieves = @people[:thieves] - officers = @people[:Officers] - if thieves <= 0 - odds_percent = 0 - elsif officers > thieves - odds_percent = 0 - else - odds = 1 \ - - officers.to_f / thieves.to_f - odds_percent = odds * 100 - end - return odds_percent - end -end \ No newline at end of file diff --git a/Gemfile b/Gemfile index 94b6f8c..6fd63b3 100644 --- a/Gemfile +++ b/Gemfile @@ -1,4 +1,4 @@ source 'https://rubygems.org' gem 'rubocop', require: false -gem 'rspec' \ No newline at end of file +gem 'rspec' diff --git a/gem_city.rb b/gem_city.rb new file mode 100644 index 0000000..9ab1362 --- /dev/null +++ b/gem_city.rb @@ -0,0 +1,43 @@ +# This class represents the town of GemCity +# This is a town riddled with crime but we can find out how happy the town is +class GemCity + attr_reader :population + attr_accessor :thieves, :officers + + def initialize + @thieves = 5 + @officers = 1 + @population = 50 + end + + def happiness_of_town + # happiness is random... people don't know what they want! + happiness_vals = (1..population).collect do + rand((100 - successful_crime_rate)..100) + end + happiness = happiness_vals.reduce(0, :+) + happiness / 100 + end + + def no_crimes_can_occur? + thieves <= 0 || officers > thieves + end + + def crime_probability + (1 - officers.to_f / thieves.to_f) * 100 + end + + def successful_crime_rate + no_crimes_can_occur? ? 0 : crime_probability + end + + def percentage_of_population(group) + format('%.0f%', (group.to_f / population.to_f) * 100.0) + end + + def city_demographics + { thieves: percentage_of_population(thieves), + officers: percentage_of_population(officers), + civilians: percentage_of_population(population - thieves - officers) } + end +end diff --git a/gem_city_spec.rb b/gem_city_spec.rb index 884b3cb..2eb3643 100644 --- a/gem_city_spec.rb +++ b/gem_city_spec.rb @@ -1,5 +1,5 @@ require 'rspec' -require_relative 'GemCity' # This line may need to be changed +require_relative 'gem_city' describe 'Gem City' do let(:city) { GemCity.new } @@ -15,19 +15,19 @@ end it 'officers = thieves' do - city.thieves 1 - city.setOfficers 1 # This line may need to be changed + city.thieves = 1 + city.officers = 1 expect(city.successful_crime_rate).to eq(0) end it 'thieves = 0' do - city.thieves 0 + city.thieves = 0 expect(city.successful_crime_rate).to eq(0) end it 'officers > thieves' do - city.thieves 1 - city.setOfficers 2 # This line may need to be changed + city.thieves = 1 + city.officers = 2 expect(city.successful_crime_rate).to eq(0) end end @@ -38,26 +38,26 @@ end it 'successful_crime_rate = 0' do - city.thieves 0 + city.thieves = 0 expect(city.happiness_of_town).to eq(50) end end - # context 'city_demographics' do - # it 'initialize' do - # demographics = city.city_demographics - # expect(demographics[:thieves]).to eq('10%') - # expect(demographics[:officers]).to eq('2%') - # expect(demographics[:civilians]).to eq('88%') - # end - - # it 'thieves = 10, officers = 25' do - # city.thieves = 10 - # city.officers = 25 - # demographics = city.city_demographics - # expect(demographics[:thieves]).to eq('20%') - # expect(demographics[:officers]).to eq('50%') - # expect(demographics[:civilians]).to eq('30%') - # end - # end + context 'city_demographics' do + it 'initialize' do + demographics = city.city_demographics + expect(demographics[:thieves]).to eq('10%') + expect(demographics[:officers]).to eq('2%') + expect(demographics[:civilians]).to eq('88%') + end + + it 'thieves = 10, officers = 25' do + city.thieves = 10 + city.officers = 25 + demographics = city.city_demographics + expect(demographics[:thieves]).to eq('20%') + expect(demographics[:officers]).to eq('50%') + expect(demographics[:civilians]).to eq('30%') + end + end end