| | """ |
| | Unit tests for the stylometry analysis service. |
| | """ |
| | import pytest |
| | from backend.app.services.stylometry import ( |
| | compute_stylometry_score, |
| | extract_features, |
| | _function_word_freq, |
| | _punctuation_pattern, |
| | _readability_score, |
| | _sentence_length_variance, |
| | ) |
| |
|
| |
|
| | class TestStylometry: |
| | def test_compute_score_normal_text(self): |
| | text = ( |
| | "The Federal Reserve announced today that it will maintain current " |
| | "interest rates through the end of the quarter, citing stable employment " |
| | "numbers and moderate inflation indicators." |
| | ) |
| | score = compute_stylometry_score(text) |
| | assert 0.0 <= score <= 1.0 |
| |
|
| | def test_compute_score_empty(self): |
| | assert compute_stylometry_score("") == 0.0 |
| |
|
| | def test_compute_score_short_text(self): |
| | assert compute_stylometry_score("Hello") == 0.0 |
| |
|
| | def test_function_word_frequency(self): |
| | text = "the dog and the cat are on the mat" |
| | ratio = _function_word_freq(text) |
| | assert 0.0 <= ratio <= 1.0 |
| | assert ratio > 0.3 |
| |
|
| | def test_punctuation_pattern(self): |
| | text = "Hello, world! How are you? I'm fine." |
| | result = _punctuation_pattern(text) |
| | assert "density" in result |
| | assert "diversity" in result |
| | assert result["density"] > 0 |
| |
|
| | def test_readability(self): |
| | text = "This is a simple test sentence. It has easy words." |
| | score = _readability_score(text) |
| | assert score >= 0 |
| |
|
| | def test_sentence_length_variance(self): |
| | text = "Short. This is a longer sentence with more words. Tiny." |
| | var = _sentence_length_variance(text) |
| | assert var >= 0 |
| |
|
| | def test_extract_features(self): |
| | text = "This is a test of the feature extraction system for analysis." |
| | features = extract_features(text) |
| | assert "function_word_ratio" in features |
| | assert "punctuation" in features |
| | assert "readability_ari" in features |
| | assert "stylometry_score" in features |
| |
|
| | def test_different_text_styles(self): |
| | formal = ( |
| | "The comprehensive analysis of macroeconomic indicators suggests that " |
| | "the prevailing monetary policy framework requires substantial revision " |
| | "in light of unprecedented fiscal challenges." |
| | ) |
| | casual = ( |
| | "yo so I was thinking maybe we should grab some food later? " |
| | "idk what u want but pizza sounds good to me lol" |
| | ) |
| | score_formal = compute_stylometry_score(formal) |
| | score_casual = compute_stylometry_score(casual) |
| | |
| | assert 0.0 <= score_formal <= 1.0 |
| | assert 0.0 <= score_casual <= 1.0 |
| |
|
