diff options
| author | Karl Lorey <git@karllorey.com> | 2020-09-26 22:12:19 +0200 |
|---|---|---|
| committer | Karl Lorey <git@karllorey.com> | 2020-09-26 22:12:19 +0200 |
| commit | ff2a5cbd711c547dfa8f43d586ba33641d8866b1 (patch) | |
| tree | 63322c2031d2a718a5d1142d26177defc8c42ec9 | |
| parent | a0606b9e7fd9319a1279fb585b351049bc2b259b (diff) | |
Add test for id-based selectors
| -rw-r--r-- | mlscraper/util.py | 4 | ||||
| -rw-r--r-- | test/test_util.py | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/mlscraper/util.py b/mlscraper/util.py index 18878b3..e24c9b0 100644 --- a/mlscraper/util.py +++ b/mlscraper/util.py @@ -46,9 +46,9 @@ def get_tree_path(node): def generate_css_selectors_for_node(node: Tag, max_classes_per_node=None): - tag_id = node.attrs.get('id', None) + tag_id = node.attrs.get("id", None) if tag_id: - yield '#' + tag_id + yield "#" + tag_id css_classes = node.attrs.get("class", []) css_class_combos = filter( diff --git a/test/test_util.py b/test/test_util.py index e35939b..09e3502 100644 --- a/test/test_util.py +++ b/test/test_util.py @@ -58,4 +58,4 @@ def test_generate_unique_path_selectors(basic_soup): assert ".wrapper.box > div" not in selectors assert "div > div" not in selectors - # assert "#sample" in selectors + assert "#sample" in selectors |