100 lines
3.8 KiB
Python
100 lines
3.8 KiB
Python
"""Tests for the examples module."""
|
|
|
|
from regex_humanizer.examples import generate_examples, generate_match_examples
|
|
|
|
|
|
class TestGenerateExamples:
|
|
"""Tests for the generate_examples function."""
|
|
|
|
def test_generate_literal_examples(self):
|
|
"""Test generating examples for a literal pattern."""
|
|
examples = generate_examples("hello", count=3)
|
|
assert len(examples) >= 1
|
|
assert "hello" in examples
|
|
|
|
def test_generate_character_class_examples(self):
|
|
"""Test generating examples for a character class."""
|
|
examples = generate_examples("[abc]", count=5)
|
|
assert len(examples) > 0
|
|
for example in examples:
|
|
assert len(example) == 1
|
|
assert example in "abc"
|
|
|
|
def test_generate_quantifier_examples(self):
|
|
"""Test generating examples for a quantifier pattern."""
|
|
examples = generate_examples("a*", count=3)
|
|
assert len(examples) >= 1
|
|
for example in examples:
|
|
assert all(c == "a" for c in example)
|
|
|
|
def test_generate_digit_examples(self):
|
|
"""Test generating examples for digit pattern."""
|
|
examples = generate_examples(r"\d+", count=3)
|
|
assert len(examples) >= 1
|
|
for example in examples:
|
|
assert example.isdigit()
|
|
|
|
def test_generate_word_examples(self):
|
|
"""Test generating examples for word character pattern."""
|
|
examples = generate_examples(r"\w+", count=3)
|
|
assert len(examples) >= 1
|
|
for example in examples:
|
|
assert example.replace("_", "").isalnum()
|
|
|
|
def test_generate_alternation_examples(self):
|
|
"""Test generating examples for alternation."""
|
|
examples = generate_examples("foo|bar", count=3)
|
|
assert len(examples) >= 1
|
|
for example in examples:
|
|
assert example in ("foo", "bar")
|
|
|
|
def test_generate_complex_pattern_examples(self):
|
|
"""Test generating examples for a complex pattern."""
|
|
examples = generate_examples(r"\d{3}-\d{4}", count=3)
|
|
assert len(examples) >= 1
|
|
for example in examples:
|
|
assert "-" in example
|
|
|
|
def test_generate_with_count(self):
|
|
"""Test that the count parameter works."""
|
|
examples = generate_examples("a", count=5)
|
|
assert len(examples) <= 5
|
|
|
|
def test_generate_invalid_pattern(self):
|
|
"""Test generating examples for an invalid pattern."""
|
|
examples = generate_examples("[unclosed", count=3)
|
|
assert examples == []
|
|
|
|
def test_generate_email_examples(self):
|
|
"""Test generating examples for an email pattern."""
|
|
examples = generate_examples(r"\w+@\w+\.\w+", count=3)
|
|
assert len(examples) >= 1
|
|
for example in examples:
|
|
assert "@" in example
|
|
assert "." in example.split("@")[1]
|
|
|
|
|
|
class TestGenerateMatchExamples:
|
|
"""Tests for the generate_match_examples function."""
|
|
|
|
def test_generate_matches_from_string(self):
|
|
"""Test generating matches from a test string."""
|
|
examples = generate_match_examples(r"\d+", "abc123def456ghi", count=3)
|
|
assert len(examples) >= 1
|
|
assert "123" in examples or "456" in examples
|
|
|
|
def test_generate_matches_no_match(self):
|
|
"""Test generating matches when no match found."""
|
|
examples = generate_match_examples(r"\d+", "abcdef", count=3)
|
|
assert examples == []
|
|
|
|
def test_generate_matches_count(self):
|
|
"""Test that count limits results."""
|
|
examples = generate_match_examples(r"\w+", "one two three four five", count=2)
|
|
assert len(examples) <= 2
|
|
|
|
def test_generate_matches_complex(self):
|
|
"""Test generating matches for complex pattern."""
|
|
examples = generate_match_examples(r"\b\w+@[\w.]+", "contact: test@example.com, support@company.org", count=3)
|
|
assert len(examples) >= 1
|