two more functions implemented

This commit is contained in:
2021-11-27 15:36:51 +01:00
parent f2abd8f2ce
commit f9745ff46d
2 changed files with 12 additions and 4 deletions

View File

@ -73,9 +73,9 @@ class FunctionsTestCase(IsolatedAsyncioTestCase):
mock_is_relevant_table_row.assert_has_calls([call(expected_output[0]), call(expected_output[1])])
def test_extract_row_data(self):
test_table = self.test_soup.find('div', attrs={'class': 'financials'}).div.div.table
test_row = self.test_soup.find('div', attrs={'class': 'financials'}).tbody.tr
expected_output = ('Item_1', (11000000, -22000000))
output = functions.extract_row_data(test_table.tbody.tr)
output = functions.extract_row_data(test_row)
self.assertTupleEqual(expected_output, output)
@patch.object(functions, 'extract_row_data')