Tests for Bankruptcy CEA #3

alifa98 · Jul 9, 2022 · 7dda527 · 7dda527
1 parent b90841f
commit 7dda527
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/gamekit/algorithms/__init__.py b/gamekit/algorithms/__init__.py
@@ -1,3 +1,5 @@
 
 
-from gamekit.algorithms import *
+from gamekit.algorithms.bankruptcy import *
+
+
diff --git a/gamekit/algorithms/bankruptcy.py b/gamekit/algorithms/bankruptcy.py
@@ -19,7 +19,7 @@ def constrained_equal_awards(claims: list, asset: float) -> tuple[list, float]:
         asset (float): total amount of our assets (non-negative)
     Returns:
         list: list of allocations
-        r: the r value in the $a_i = min(r, c_i) $ (if sum of claims become less than asset, then the r-value would be the asset itself.)
+        r: the r value in the $a_i = min(r, c_i) $ (if sum of claims become less than asset, then the r-value would be the largest claim value.)
     """
 
     # checks
@@ -30,7 +30,7 @@ def constrained_equal_awards(claims: list, asset: float) -> tuple[list, float]:
         raise NegativeNumberException("Asset cannot be a negative number.")
 
     if sum(claims) <= asset:
-        return claims, asset
+        return claims, max(claims)
     else:
 
         equal_alloc = asset/len(claims)

diff --git a/gamekit/exceptions.py b/gamekit/exceptions.py
@@ -6,5 +6,3 @@ class NegativeNumberException(Exception):
     def __init__(self, message) -> None:
         self.message = message
 
-    def __init__(self) -> None:
-        pass