Broke up some monolithic unit test files.

1 files changed, 18 insertions, 0 deletions

diff --git a/bs4/tests/__init__.py b/bs4/tests/__init__.py
index 97914c0..6b70c34 100644
--- a/bs4/tests/__init__.py
+++ b/bs4/tests/__init__.py
@@ -230,6 +230,24 @@ class SoupTest(object):
             # Return the child to the recursive caller
             return child
 
+    def assert_selects(self, tags, should_match):
+        """Make sure that the given tags have the correct text.
+
+        This is used in tests that define a bunch of tags, each
+        containing a single string, and then select certain strings by
+        some mechanism.
+        """
+        assert [tag.string for tag in tags] == should_match
+
+    def assert_selects_ids(self, tags, should_match):
+        """Make sure that the given tags have the correct IDs.
+
+        This is used in tests that define a bunch of tags, each
+        containing a single string, and then select certain strings by
+        some mechanism.
+        """
+        assert [tag['id'] for tag in tags] == should_match
+
 
 class TreeBuilderSmokeTest(object):
     # Tests that are common to HTML and XML tree builders.