from unittest import TestCase
from parsing import SpreadSheetParser


class PdfParserTest(TestCase):

    def test_parse_content_xls(self):

        parser = SpreadSheetParser([], 1500)

        info = parser.parse("test_files/xls1.xls")

        self.assertEqual(len(info["content"]), 1500)

    def test_parse_content_xlsx(self):

        parser = SpreadSheetParser([], 1500)

        info = parser.parse("test_files/xlsx1.xlsx")

        self.assertEqual(len(info["content"]), 1500)