fix straights tests

Author: HighDiceRoller

tests/deck_test.py

@@ -26,7 +26,7 @@ def test_empty_deal():
 
 def test_poker_straight():
     deal = icepool.Deck(range(13), times=4).deal(5)
-    result = best_run_evaluator.evaluate(deal).marginals[0] == 5
+    result = best_run_evaluator.evaluate(deal) == 5
     assert result.equals(icepool.coin(9216, 2598960))
 
 

tests/evaluator_test.py

@@ -79,7 +79,8 @@ def test_standard_pool_zero_dice():
 
 
 def test_runs():
-    result = icepool.BestStraightEvaluator()(icepool.standard_pool([12, 10, 8]))
+    result = icepool.BestStraightEvaluator(include_outcome=True)(
+        icepool.standard_pool([12, 10, 8]))
 
     def func(*outcomes):
         outcomes = sorted(outcomes)
@@ -100,7 +101,7 @@ def func(*outcomes):
 
 def test_runs_skip():
     die = icepool.Die([0, 10])
-    result = icepool.BestStraightEvaluator()(die.pool(10))
+    result = icepool.BestStraightEvaluator(include_outcome=True)(die.pool(10))
     assert result.outcomes() == ((1, 0), (1, 10))