diff --git a/src/breadability/tests/test_readable.py b/src/breadability/tests/test_readable.py index f8a4ab2..93889d9 100644 --- a/src/breadability/tests/test_readable.py +++ b/src/breadability/tests/test_readable.py @@ -185,7 +185,7 @@ class TestScoringNodes(TestCase): # this might change as we tweak our algorithm, but if it does change, # it signifies we need to look at what we changed. - self.assertEqual(len(candidates.keys()), 8) + self.assertEqual(len(candidates.keys()), 6) # one of these should have a decent score scores = sorted([c.content_score for c in candidates.values()]) diff --git a/src/breadability/tests/test_scoring.py b/src/breadability/tests/test_scoring.py index 905867a..72f185f 100644 --- a/src/breadability/tests/test_scoring.py +++ b/src/breadability/tests/test_scoring.py @@ -1,3 +1,5 @@ +import re + from lxml.html import fragment_fromstring from unittest import TestCase @@ -13,30 +15,30 @@ class TestCheckNodeAttr(TestCase): """ def test_has_class(self): """Verify that a node has a class in our set.""" - test_set = set(['test1', 'test2']) + test_re = re.compile('test1|test2', re.I) test_node = fragment_fromstring('
') test_node.set('class', 'test2 comment') - self.assertTrue(check_node_attr(test_node, 'class', test_set)) + self.assertTrue(check_node_attr(test_node, 'class', test_re)) def test_has_id(self): """Verify that a node has an id in our set.""" - test_set = set(['test1', 'test2']) + test_re = re.compile('test1|test2', re.I) test_node = fragment_fromstring('
') test_node.set('id', 'test2') - self.assertTrue(check_node_attr(test_node, 'id', test_set)) + self.assertTrue(check_node_attr(test_node, 'id', test_re)) def test_lacks_class(self): """Verify that a node does not have a class in our set.""" - test_set = set(['test1', 'test2']) + test_re = re.compile('test1|test2', re.I) test_node = fragment_fromstring('
') test_node.set('class', 'test4 comment') - self.assertFalse(check_node_attr(test_node, 'class', test_set)) + self.assertFalse(check_node_attr(test_node, 'class', test_re)) def test_lacks_id(self): """Verify that a node does not have an id in our set.""" - test_set = set(['test1', 'test2']) + test_re = re.compile('test1|test2', re.I) test_node = fragment_fromstring('
') test_node.set('id', 'test4') - self.assertFalse(check_node_attr(test_node, 'id', test_set)) + self.assertFalse(check_node_attr(test_node, 'id', test_re))