diff --git a/tests/test_preprocessing.py b/tests/test_preprocessing.py
index 208bb5b..0a57873 100644
--- a/tests/test_preprocessing.py
+++ b/tests/test_preprocessing.py
@@ -2,6 +2,7 @@ import pytest
 import librosa
 from src.preprocessing.normalize import normalize_audio
 from src.preprocessing.trim import trim_audio
+from src.preprocessing.classify import classify_audio
 from src.input.file_reader import read_audio
 
 def test_normalize_audio():
@@ -17,4 +18,11 @@ def test_trim_audio():
     audio, sr = read_audio(file_path)
     trimmed_audio = trim_audio(audio, sr)
 
-    assert len(trimmed_audio) <= len(audio)
\ No newline at end of file
+    assert len(trimmed_audio) <= len(audio)
+
+def test_classify():
+    file_path = "samples/cafe_crowd_talk.wav"
+    expected_class = "Speech"
+    predicted_class = classify_audio(file_path)
+
+    assert predicted_class == expected_class , f"Expected {expected_class}, but got {predicted_class}"
\ No newline at end of file