pretty print methods should return the string, not just print

ihendley · May 31, 2018 · f7505e6 · f7505e6
1 parent c9cdcee
commit f7505e6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/treys/card.py b/treys/card.py
@@ -196,7 +196,7 @@ def print_pretty_card(card_int):
         """
         Expects a single integer as input
         """
-        print(Card.int_to_pretty_str(card_int))
+        return Card.int_to_pretty_str(card_int)
 
     @staticmethod
     def print_pretty_cards(card_ints):
@@ -211,4 +211,4 @@ def print_pretty_cards(card_ints):
             else:
                 output += str(Card.int_to_pretty_str(c)) + " "
 
-        print(output)
+        return output