Refactored unit tests

This commit is contained in:
simon987
2018-04-21 22:43:38 -04:00
parent ef168a5cab
commit 75d7e8e3a8
73 changed files with 165 additions and 145 deletions

View File

@@ -0,0 +1,25 @@
from unittest import TestCase
from parsing import SpreadSheetParser
import os
dir_name = os.path.dirname(os.path.abspath(__file__))
class PdfParserTest(TestCase):
def test_parse_content_xls(self):
parser = SpreadSheetParser([], 1500, "test_files/")
info = parser.parse(dir_name + "/test_files/xls1.xls")
self.assertEqual(len(info["content"]), 1500)
def test_parse_content_xlsx(self):
parser = SpreadSheetParser([], 1500, "test_files/")
info = parser.parse(dir_name + "/test_files/xlsx1.xlsx")
self.assertEqual(len(info["content"]), 1500)