Fix bug with Game.write for 'gte' format

gambitproject · Nov 9, 2023 · 3ff06e2 · 3ff06e2
1 parent 90bbf7c
commit 3ff06e2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/pygambit/gte.py b/src/pygambit/gte.py
@@ -72,10 +72,10 @@ def read_game(f):
 
 
 def write_game_outcome(game, outcome, doc, xml_parent):
-    for (i, p) in enumerate(game.players):
+    for p in game.players:
         if outcome is not None:
             etree.SubElement(xml_parent, "payoff",
-                             player=p.label).text = str(outcome[i])
+                             player=p.label).text = str(outcome[p])
         else:
             etree.SubElement(xml_parent, "payoff",
                              player=p.label).text = "0"

diff --git a/src/pygambit/tests/test_file.py b/src/pygambit/tests/test_file.py
@@ -73,6 +73,10 @@ def test_parse_string_extra_payoff(self):
             "Parse error in game file: line 5:29: Expecting '}' after outcome"
         )
 
+    def test_write_game_gte_sanity(self):
+        g = pygambit.Game.parse_game(self.file_text)
+        g.write('gte')
+
 
 class TestGambitNfgFile(unittest.TestCase):
     def setUp(self):