Peter M. Groen / oletools

Commit 2e672973b29fba804f10a34f05d2207ddcce762e

Authored by Christian Herdtweck
1 parent 140e3608

tests: Treat xlsb properly in ftguess tests

Inline Side-by-side
Showing 1 changed file with 1 additions and 6 deletions
tests/ftguess/test_basic.py
  View file @2e67297
@@ -22,7 +22,7 @@ class TestFTGuess(unittest.TestCase): @@ -22,7 +22,7 @@ class TestFTGuess(unittest.TestCase):
22 ftguess.FType_Word2007, ftguess.FType_Word2007_Macro, 22 ftguess.FType_Word2007, ftguess.FType_Word2007_Macro,
23 ftguess.FType_Word2007_Template, 23 ftguess.FType_Word2007_Template,
24 ftguess.FType_Word2007_Template_Macro, ftguess.FType_Excel97, 24 ftguess.FType_Word2007_Template_Macro, ftguess.FType_Excel97,
25 - ftguess.FType_Excel2007, 25 + ftguess.FType_Excel2007, ftguess.FType_Excel2007_XLSB,
26 ftguess.FType_Excel2007_XLSX , ftguess.FType_Excel2007_XLSM , 26 ftguess.FType_Excel2007_XLSX , ftguess.FType_Excel2007_XLSM ,
27 ftguess.FType_Excel2007_Template, 27 ftguess.FType_Excel2007_Template,
28 ftguess.FType_Excel2007_Template_Macro, 28 ftguess.FType_Excel2007_Template_Macro,
@@ -38,9 +38,6 @@ class TestFTGuess(unittest.TestCase): @@ -38,9 +38,6 @@ class TestFTGuess(unittest.TestCase):
38 for extension in ftype.extensions: 38 for extension in ftype.extensions:
39 ftype_for_extension[extension] = ftype 39 ftype_for_extension[extension] = ftype
40 40
41 - # TODO: xlsb is not implemented yet  
42 - ftype_for_extension['xlsb'] = ftguess.FType_Generic_OpenXML  
43 -  
44 for filename, file_contents in loop_over_files(): 41 for filename, file_contents in loop_over_files():
45 # let the system guess 42 # let the system guess
46 guess = ftguess.ftype_guess(data=file_contents) 43 guess = ftguess.ftype_guess(data=file_contents)
@@ -94,9 +91,7 @@ class TestFTGuess(unittest.TestCase): @@ -94,9 +91,7 @@ class TestFTGuess(unittest.TestCase):
94 if expect not in (ftguess.FType_Generic_OLE, ftguess.FType_Unknown): 91 if expect not in (ftguess.FType_Generic_OLE, ftguess.FType_Unknown):
95 self.assertEqual(guess.is_excel(), extension.startswith('x') 92 self.assertEqual(guess.is_excel(), extension.startswith('x')
96 and extension != 'xml' 93 and extension != 'xml'
97 - and extension != 'xlsb'  
98 and extension != 'xps') 94 and extension != 'xps')
99 - # xlsb is excel but not handled properly yet  
100 self.assertEqual(guess.is_word(), extension.startswith('d')) 95 self.assertEqual(guess.is_word(), extension.startswith('d'))
101 self.assertEqual(guess.is_powerpoint(), 96 self.assertEqual(guess.is_powerpoint(),
102 extension.startswith('p')) 97 extension.startswith('p'))