Sentiment-Analysis-Chatbot/test_project.py at main · harshitt13/Sentiment-Analysis-Chatbot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
import pytest
from project import analyze_sentiment, get_response, format_message, save_conversation
import json
import os

def test_analyze_sentiment():
    # Test positive sentiment
    assert analyze_sentiment("I love this!")["category"] == "very positive"

    # Test negative sentiment
    assert analyze_sentiment("I hate this.")["category"] == "very negative"

    # Test neutral sentiment
    assert analyze_sentiment("The sky is blue.")["category"] == "neutral"

    # Test empty string
    with pytest.raises(ValueError):
        analyze_sentiment("")

    # Test non-string input
    with pytest.raises(ValueError):
        analyze_sentiment(123)

    # Test sentiment structure
    result = analyze_sentiment("Test message")
    assert "polarity" in result
    assert "subjectivity" in result
    assert "category" in result
    assert isinstance(result["polarity"], float)
    assert isinstance(result["subjectivity"], float)
    assert isinstance(result["category"], str)

def test_get_response():
    # Test all valid categories
    assert isinstance(get_response("very positive"), str)
    assert isinstance(get_response("positive"), str)
    assert isinstance(get_response("neutral"), str)
    assert isinstance(get_response("negative"), str)
    assert isinstance(get_response("very negative"), str)

    # Test invalid category
    with pytest.raises(ValueError):
        get_response("invalid_category")

    # Test non-string input
    with pytest.raises(ValueError):
        get_response(123)

    # Test case insensitivity
    assert get_response("VERY POSITIVE") == get_response("very positive")

def test_format_message():
    # Test basic message formatting
    result = format_message("User", "Hello")
    assert isinstance(result, dict)
    assert "timestamp" in result
    assert "speaker" in result
    assert "message" in result
    assert result["speaker"] == "User"
    assert result["message"] == "Hello"

    # Test with sentiment
    sentiment = {"category": "positive", "polarity": 0.5, "subjectivity": 0.5}
    result = format_message("User", "Hello", sentiment)
    assert "sentiment" in result
    assert result["sentiment"] == sentiment

    # Test invalid inputs
    with pytest.raises(ValueError):
        format_message(123, "message")
    with pytest.raises(ValueError):
        format_message("User", 123)

def test_save_conversation():
    # Test valid conversation save
    conversation = [
        {"timestamp": "2024-03-21 10:00:00", "speaker": "User", "message": "Hello"},
        {"timestamp": "2024-03-21 10:00:01", "speaker": "Bot", "message": "Hi"}
    ]
    assert save_conversation(conversation, "test_chat.json") == True

    # Verify file contents
    with open("test_chat.json", "r") as f:
        saved_data = json.load(f)
    assert saved_data == conversation

    # Clean up test file
    os.remove("test_chat.json")

    # Test invalid input
    with pytest.raises(ValueError):
        save_conversation("not a list")

    # Test invalid file path
    with pytest.raises(IOError):
        save_conversation(conversation, "/invalid/path/chat.json")