#!/usr/bin/python
# -*- coding: utf-8 -*-
import sys
import unittest
# if sys.version_info[0] < 3:
# import mock
# else:
# from unittest import mock
from bs4 import BeautifulSoup
from html_table_extractor.extractor import Extractor
class TestSimpleExtractor(unittest.TestCase):
def setUp(self):
html = """
"""
self.extractor = Extractor(html)
self.extractor.parse()
def test_return_list(self):
self.assertEqual(
self.extractor.return_list(),
[[u'1', u'2'], [u'3', u'4']]
)
class TestExtractorTransformer(unittest.TestCase):
def setUp(self):
html = """
"""
self.extractor = Extractor(html, transformer=int)
self.extractor.parse()
def test_config_transformer(self):
self.assertEqual(
self.extractor.return_list(),
[[1, 2], [3, 4]]
)
class TestPassId(unittest.TestCase):
def test_init_with_id(self):
html = """
"""
soup = BeautifulSoup(html, 'html.parser')
extractor = Extractor(soup, id_='wanted').parse()
self.assertEqual(
extractor.return_list(),
[[u'1', u'2'], [u'3', u'4']]
)
class TestComplexExtractor(unittest.TestCase):
def setUp(self):
html = """
"""
self.extractor = Extractor(html)
self.extractor.parse()
def test_return_list(self):
self.assertEqual(
self.extractor.return_list(),
[[u'1', u'2', u'3'], [u'1', u'4', u'4'], [u'5', u'5', u'5']]
)
class TestConflictedExtractor(unittest.TestCase):
def setUp(self):
html = """
"""
self.extractor = Extractor(html)
self.extractor.parse()
def test_return_list(self):
self.assertEqual(
self.extractor.return_list(),
[[u'1', u'2', u'3'], [u'1', u'4', u'3'], [u'5', u'5', u'3']]
)
# @mock.patch('csv.writer')
# @mock.patch('html_table_extractor.extractor.open')
# class TestWriteToCsv(unittest.TestCase):
# def setUp(self):
# html = """
#
#
# | 1 |
# 2 |
#
#
# | 3 |
# 4 |
#
#
# """
# self.extractor = Extractor(html)
# self.extractor.parse()
# mock.mock_open()
#
# def test_write_to_csv_default(self, csv_mock, _):
# self.extractor.write_to_csv()
# csv_mock.assert_called_with('./output.csv', 'w')
#
# def test_write_to_csv_custom_path_and_filename(self, csv_mock, _):
# self.extractor.write_to_csv(path='/test/path', filename='testfile.csv')
# csv_mock.assert_called_with('/test/path/testfile.csv', 'w')
if __name__ == '__main__':
unittest.main()