| """Uji modul AI: deteksi spam & klasifikasi kategori (revisi UC-02 & UC-04).""" |
|
|
| from src.constants import CATEGORIES, LABEL_LAYAK, LABEL_SPAM |
|
|
|
|
| def test_spam_terdeteksi(classifier): |
| res = classifier.predict("asdkjasd test test 123 wkwkwk iseng doang") |
| assert res.is_spam is True |
| assert res.spam_label == LABEL_SPAM |
| assert res.kategori is None |
|
|
|
|
| def test_laporan_layak_tidak_dianggap_spam(classifier): |
| res = classifier.predict( |
| "Jalan di depan rumah saya berlubang besar dan membahayakan pengendara" |
| ) |
| assert res.is_spam is False |
| assert res.spam_label == LABEL_LAYAK |
| assert res.kategori in CATEGORIES |
|
|
|
|
| def test_klasifikasi_kategori_infrastruktur(classifier): |
| res = classifier.predict("Lampu jalan mati dan jalan berlubang membahayakan pengendara") |
| assert res.kategori == "Infrastruktur" |
| assert 0.0 <= res.kategori_confidence <= 1.0 |
|
|
|
|
| def test_klasifikasi_kategori_lingkungan(classifier): |
| res = classifier.predict("Sampah menumpuk di sungai menyebabkan banjir dan bau busuk") |
| assert res.kategori == "Lingkungan" |
|
|
|
|
| def test_teks_kosong_dianggap_spam(classifier): |
| res = classifier.predict(" ") |
| assert res.is_spam is True |
| assert res.kategori is None |
|
|
|
|
| def test_confidence_dalam_rentang_valid(classifier): |
| res = classifier.predict("Pelayanan KTP di kecamatan sangat lambat dan berbelit") |
| assert 0.0 <= res.spam_confidence <= 1.0 |
| assert 0.0 <= res.kategori_confidence <= 1.0 |
|
|
