add players at the median score of the league

bytes-zone · Oct 12, 2020 · a52ebb8 · a52ebb8
1 parent 97e2116
commit a52ebb8
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/src/League.elm b/src/League.elm
@@ -92,13 +92,25 @@ getPlayer name (League league) =
 
 addPlayer : Player -> League -> League
 addPlayer player (League league) =
-    League { league | players = Dict.insert player.name player league.players }
+    let
+        ratings =
+            Dict.values league.players
+                |> List.map .rating
+                |> List.sort
+
+        medianishRating =
+            ratings
+                |> List.drop (List.length ratings // 2)
+                |> List.head
+                |> Maybe.withDefault Elo.initialRating
+    in
+    League { league | players = Dict.insert player.name (Player.setRating medianishRating player) league.players }
 
 
 {-| -}
 updatePlayer : Player -> League -> League
-updatePlayer =
-    addPlayer
+updatePlayer player (League league) =
+    League { league | players = Dict.insert player.name player league.players }
 
 
 retirePlayer : Player -> League -> League

diff --git a/tests/LeagueTest.elm b/tests/LeagueTest.elm
@@ -60,7 +60,8 @@ playersTests =
                 League.init
                     |> League.addPlayer player
                     |> League.players
-                    |> Expect.equal [ player ]
+                    |> List.map .name
+                    |> Expect.equal [ player.name ]
         , fuzz playerFuzzer "retiring a player removes them from the players list" <|
             \player ->
                 League.init
@@ -109,7 +110,8 @@ startMatchTests =
                     |> League.addPlayer playerB
                     |> League.startMatch (Match playerA playerB)
                     |> League.currentMatch
-                    |> Expect.equal (Just (Match playerA playerB))
+                    |> Maybe.map (\(Match a b) -> ( a.name, b.name ))
+                    |> Expect.equal (Just ( playerA.name, playerB.name ))
         , fuzz playerFuzzer "you can't start a match with one player against themselves" <|
             \player ->
                 League.init