Merge pull request #26 from Metta-AI/andre-kinship-3

Metta-AI · Jan 13, 2025 · 1f3c8d7 · 1f3c8d7
2 parents 49de780 + 2fe1600
commit 1f3c8d7
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/test_kinship.py b/tests/test_kinship.py
@@ -86,9 +86,23 @@ def test_shared_rewards(msg, rewards, expected, team_reward):
   team_reward = 0.5
 )
 
+test_shared_rewards(
+  msg = "1 reward split 1/3%",
+  rewards = [1, 0, 0, 0],
+  expected = [0.8333334, 0.16666667, 0, 0],
+  team_reward = 1/3
+)
+
 test_shared_rewards(
   msg = "2 rewards split 50%",
   rewards = [1, 1, 0, 0],
   expected = [1, 1, 0, 0],
   team_reward = 0.5
 )
+
+test_shared_rewards(
+  msg = "1 rewards no split",
+  rewards = [1, 0, 0, 0],
+  expected = [1, 0, 0, 0],
+  team_reward = 0.0
+)