fix up tests

bytes-zone · Sep 15, 2020 · 8bd8c0e · 8bd8c0e
1 parent 09cf006
commit 8bd8c0e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/src/League.elm b/src/League.elm
@@ -1,7 +1,7 @@
 module League exposing
     ( League, init, decoder, encode
     , addPlayer, players, getPlayer, retirePlayer
-    , Match(..), currentMatch, nextMatch, startMatch, Outcome(..), finishMatch
+    , Match(..), currentMatch, nextMatch, startMatch, Outcome(..), finishMatch, kFactor
     )
 
 {-|
@@ -10,7 +10,7 @@ module League exposing
 
 @docs addPlayer, players, getPlayer, retirePlayer
 
-@docs Match, currentMatch, nextMatch, startMatch, Outcome, finishMatch
+@docs Match, currentMatch, nextMatch, startMatch, Outcome, finishMatch, kFactor
 
 -}
 
@@ -300,8 +300,6 @@ playInMatches =
     5
 
 
-{-| Chesterton's export
--}
 kFactor : Player -> Int
 kFactor player =
     if player.matches < playInMatches then

diff --git a/tests/LeagueTest.elm b/tests/LeagueTest.elm
@@ -157,7 +157,7 @@ finishMatchTests =
                 \winner ->
                     let
                         newRatings =
-                            Elo.win Elo.sensitiveKFactor
+                            Elo.win (League.kFactor winner)
                                 { won = winner.rating
                                 , lost = dummy.rating
                                 }
@@ -204,7 +204,15 @@ finishMatchTests =
                 \player ->
                     let
                         newRatings =
-                            Elo.draw Elo.sensitiveKFactor
+                            Elo.draw
+                                (League.kFactor
+                                    (if player.rating > dummy.rating then
+                                        player
+
+                                     else
+                                        dummy
+                                    )
+                                )
                                 { playerA = player.rating
                                 , playerB = dummy.rating
                                 }