#!/usr/bin/env python

# -*- coding: utf-8 -*-

from runner.koan import *

# Greed is a dice game where you roll up to five dice to accumulate

# points. The following "score" function will be used calculate the

# score of a single roll of the dice.

#

# A greed roll is scored as follows:

#

# * A set of three ones is 1000 points

#

# * A set of three numbers (other than ones) is worth 100 times the

# number. (e.g. three fives is 500 points).

#

# * A one (that is not part of a set of three) is worth 100 points.

#

# * A five (that is not part of a set of three) is worth 50 points.

#

# * Everything else is worth 0 points.

#

#

# Examples:

#

# score([1,1,1,5,1]) => 1150 points

# score([2,3,4,6,2]) => 0 points

# score([3,4,5,3,3]) => 350 points

# score([1,5,1,2,4]) => 250 points

#

# More scoring examples are given in the tests below:

#

# Your goal is to write the score method.

def score(dice):

score = 0

tallies = {}.fromkeys(set(dice), 0)

for die in dice:

tallies[die] += 1

if 1 in tallies and tallies[1] >= 3:

tallies[1] -= 3

score += 1000

for k in tallies:

if tallies[k] >= 3:

tallies[k] -= 3

score += k * 100

for k in tallies:

if k == 1:

score += 100 * tallies[k]

elif k== 5:

score += 50 * tallies[k]

return score

##### # You need to write this method

##### pass

class AboutScoringProject(Koan):

def test_score_of_an_empty_list_is_zero(self):

self.assertEqual(0, score([]))

def test_score_of_a_single_roll_of_5_is_50(self):

self.assertEqual(50, score([5]))

def test_score_of_a_single_roll_of_1_is_100(self):

self.assertEqual(100, score([1]))

def test_score_of_multiple_1s_and_5s_is_the_sum_of_individual_scores(self):

self.assertEqual(300, score([1,5,5,1]))

def test_score_of_single_2s_3s_4s_and_6s_are_zero(self):

self.assertEqual(0, score([2,3,4,6]))

def test_score_of_a_triple_1_is_1000(self):

self.assertEqual(1000, score([1,1,1]))

def test_score_of_other_triples_is_100x(self):

self.assertEqual(200, score([2,2,2]))

self.assertEqual(300, score([3,3,3]))

self.assertEqual(400, score([4,4,4]))

self.assertEqual(500, score([5,5,5]))

self.assertEqual(600, score([6,6,6]))

def test_score_of_mixed_is_sum(self):

self.assertEqual(250, score([2,5,2,2,3]))

self.assertEqual(550, score([5,5,5,5]))