Add sample name to unittest fail message.

Should help debugging unexpected results.
decalage2 · May 31, 2024 · 266b0e0 · 266b0e0
1 parent 6b18caa
commit 266b0e0
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tests/oleid/test_basic.py b/tests/oleid/test_basic.py
@@ -68,7 +68,7 @@ def test_properties(self):
                 self.assertEqual(value_dict['author'],
                                  b'\xb1\xe8\xb1\xe2\xc1\xa4;kijeong')
             elif 'olevba/sample_with_vba.ppt' in filename:
-                print("\nTODO: find reason for different results for olevba/sample_with_vba.ppt")
+                print('\nTODO: find reason for different results for olevba/sample_with_vba.ppt')
                 # on korean test machine, this is the result:
                 # self.assertEqual(value_dict['codepage'],
                 #                  '949: ANSI/OEM Korean (Unified Hangul Code)')
@@ -77,10 +77,14 @@ def test_properties(self):
                 continue
             else:
                 self.assertEqual(value_dict['codepage'],
-                                 '1252: ANSI Latin 1; Western European (Windows)')
+                                 '1252: ANSI Latin 1; Western European (Windows)',
+                                 'Unexpected result {0!r} for codepage of sample {1}'
+                                 .format(value_dict['codepage'], filename))
                 self.assertIn(value_dict['author'],
                               (b'user', b'schulung',
-                               b'xxxxxxxxxxxx', b'zzzzzzzzzzzz'))
+                               b'xxxxxxxxxxxx', b'zzzzzzzzzzzz'),
+                              'Unexpected result {0!r} for author of sample {1}'
+                              .format(value_dict['author'], filename))
 
     def test_encrypted(self):
         """Test indicator "encrypted"."""