| import unittest | |
| from details_data_processor import DetailsDataProcessor | |
| import pandas as pd | |
| import requests | |
| import os | |
| class TestDetailsDataProcessor(unittest.TestCase): | |
| def setUp(self): | |
| self.processor = DetailsDataProcessor() | |
| # check that the result is a pandas dataframe | |
| def test_process_data(self): | |
| pass | |
| # data = self.processor.data | |
| # self.assertIsInstance(data, pd.DataFrame) | |
| def test_download_file(self): | |
| DetailsDataProcessor._download_file('https://www.google.com', 'test.html') | |
| self.assertTrue(os.path.exists('test.html')) | |
| os.remove('test.html') | |
| if __name__ == '__main__': | |
| unittest.main() |