100% coverage

2021-12-26 23:32:15 +01:00
parent a2f3f590f6
commit f29f22a52e
3 changed files with 6 additions and 7 deletions
--- a/tests/test_functions.py
+++ b/tests/test_functions.py
@ -52,8 +52,11 @@ class FunctionsTestCase(IsolatedAsyncioTestCase):
        self.assertEqual(expected_output, output)

    def test_get_row_indent(self):
-        trs = self.test_soup.find_all('tr')
+        mock_row = BeautifulSoup('<tr><div>foo</div></tr>', HTML_PARSER).tr
        expected_output = 0
+        output = functions.get_row_indent(mock_row)
+        self.assertEqual(expected_output, output)
+        trs = self.test_soup.find_all('tr')
        output = functions.get_row_indent(trs[0])
        self.assertEqual(expected_output, output)
        for i, tr in enumerate(trs[1:], start=1):