add unit tests

Author: nkansal96

tests/assets/hw.wav

@@ -0,0 +1,107 @@
+import json, pytest
+from auroraapi.api import *
+from auroraapi.globals import _config
+
+class MockResponse(object):
+	def __init__(self):
+		self.status_code = 200
+		self.headers = {}
+		self.text = ""
+	
+	def json(self):
+		return json.loads(self.text)
+
+class TestAPIException(object):
+	def test_create(self):
+		e = APIException("id","status","code","type","message")
+		assert isinstance(e, APIException)
+		assert e.id == "id"
+		assert e.status == "status"
+		assert e.code == "code"
+		assert e.type == "type"
+		assert e.message == "message"
+	
+	def test_str(self):
+		e = APIException("id","status","code","type","message")
+		assert str(e) == "[{}] {}".format(e.code, e.message)
+
+	def test_str_no_code(self):
+		e = APIException("id","status",None,"type","message")
+		assert str(e) == "[{}] {}".format(e.status, e.message)
+
+	def test_repr(self):
+		e = APIException("id","status","code","type","message")
+		j = json.loads(repr(e))
+		assert j["id"] == e.id
+		assert j["status"] == e.status
+		assert j["code"] == e.code
+		assert j["type"] == e.type
+		assert j["message"] == e.message
+
+class TestAPIUtils(object):
+	def setup(self):
+		_config.app_id = "appid"
+		_config.app_token = "apptoken"
+	
+	def teardown(self):
+		_config.app_id = None
+		_config.app_token = None
+	
+	def test_get_headers(self):
+		h = get_headers()
+		assert h["X-Application-ID"] == _config.app_id
+		assert h["X-Application-Token"] == _config.app_token
+		assert h["X-Device-ID"] == _config.device_id
+	
+	def test_handle_error_no_error(self):
+		handle_error(MockResponse())
+	
+	def test_handle_error_json(self):
+		r = MockResponse()
+		r.status_code = 400
+		r.headers = { "content-type": "application/json" }
+		r.text = json.dumps({
+			"id": "id",
+			"code": "MissingApplicationIDHeader",
+			"type": "BadRequest",
+			"status": 400,
+			"message": "message"
+		})
+
+		with pytest.raises(APIException) as e:
+			handle_error(r)
+			assert e.id == "id"
+			assert e.code == "MissingApplicationIDHeader"
+			assert e.type == "BadRequest"
+			assert e.status == 400
+			assert e.message == "message"
+	
+	def test_handle_error_413(self):
+		r = MockResponse()
+		r.status_code = 413
+		r.headers["content-type"] = "text/html"
+		r.text = "Request entity too large"
+
+		with pytest.raises(APIException) as e:
+			handle_error(r)
+			assert e.id == None
+			assert e.status == 413
+			assert e.type == "RequestEntityTooLarge"
+			assert e.code == "RequestEntityTooLarge"
+			assert e.message == "Request entity too large"
+	
+	def test_handle_error_other(self):
+		r = MockResponse()
+		r.status_code = 503
+		r.headers["content-type"] = "text/html"
+		r.text = "Service unavailable"
+
+
+		with pytest.raises(APIException) as e:
+			handle_error(r)
+			assert e.id == None
+			assert e.status == 413
+			assert e.type == None
+			assert e.code == None
+			assert e.message == r.text
+			assert str(e) == "[{}] {}".format(r.status_code, r.text)

tests/test_api.py

@@ -0,0 +1,79 @@
+import pytest, os, tempfile
+from auroraapi.audio import *
+
+class TestAudioFile(object):
+	def test_create(self):
+		with open("tests/assets/hw.wav", "rb") as f:
+			d = f.read()
+			a = AudioFile(d)
+			assert isinstance(a, AudioFile)
+			assert len(a.audio.raw_data) + 44 == len(d)
+	
+	def test_write_to_file(self):
+		path = os.path.join(tempfile.gettempdir(), "out.wav")
+		with open("tests/assets/hw.wav", "rb") as f:
+			d1 = f.read()
+			a1 = AudioFile(d1)
+			a1.write_to_file(path)
+
+			with open(path, "rb")	as f:
+				d2 = f.read()
+				a2 = AudioFile(d2)
+				assert d2 == d1
+				assert a1.audio == a2.audio
+
+		os.remove(path)
+
+	def test_get_wav(self):
+		with open("tests/assets/hw.wav", "rb") as f:
+			d = f.read()
+			a = AudioFile(d)
+			w = a.get_wav()
+			assert d == w
+	
+	def test_pad(self):
+		with open("tests/assets/hw.wav", "rb") as f:
+			d = f.read()
+			a = AudioFile(d)
+			# adds 1s of padding to each side (2 x (16000 Hz * 16 bits/sample * 1 byte/8bits))
+			pa = a.pad(1)
+			w = pa.get_wav()
+			assert len(w) == len(d) + 64000
+			assert w[-1] == 0
+			assert w[44] == 0
+			assert w[44 + 32000] == d[44]
+
+	def test_pad_left(self):
+		with open("tests/assets/hw.wav", "rb") as f:
+			d = f.read()
+			a = AudioFile(d)
+			# adds 1s of padding to left side (1 x (16000 Hz * 16 bits/sample * 1 byte/8bits))
+			pa = a.pad_left(1)
+			w = pa.get_wav()
+			assert len(w) == len(d) + 32000
+			assert w[-1] == d[-1]
+			assert w[44] == 0
+			assert w[44 + 32000] == d[44]
+	
+	def test_pad_right(self):
+		with open("tests/assets/hw.wav", "rb") as f:
+			d = f.read()
+			a = AudioFile(d)
+			# adds 1s of padding to right side (1 x (16000 Hz * 16 bits/sample * 1 byte/8bits))
+			pa = a.pad_right(1)
+			w = pa.get_wav()
+			assert len(w) == len(d) + 32000
+			assert w[-1] == 0
+			assert w[44] == d[44]
+			assert w[-32000] == 0
+	
+	def test_trim_silent(self):
+		with open("tests/assets/hw.wav", "rb") as f:
+			d = f.read()
+			a = AudioFile(d)
+			t = a.trim_silent()
+			# TODO: actually add a test here. for now, it just checks for compilation
+	
+	# def test_play(self):
+
+

tests/test_audio.py

@@ -0,0 +1,16 @@
+from auroraapi.globals import Config
+
+class TestGlobals(object):
+	def test_create_config(self):
+		c = Config()
+		assert isinstance(c, Config)
+		assert c != None
+	
+	def test_assign_config(self):
+		c = Config()
+		c.app_id = "test"
+		c.app_token = "test123"
+
+		assert c.app_id == "test"
+		assert c.app_token == "test123"
+		assert c.device_id == None

tests/test_globals.py

@@ -0,0 +1,25 @@
+import pytest
+from auroraapi.interpret import Interpret
+
+class TestInterpret(object):
+	def test_create_no_arguments(self):
+		with pytest.raises(TypeError):
+			Interpret()
+	
+	def test_create_wrong_type(self):
+		with pytest.raises(TypeError):
+			Interpret("test")
+	
+	def test_create(self):
+		d = { "intent": "test", "entities": {} }
+		i = Interpret(d)
+		assert isinstance(i, Interpret)
+		assert i.intent == "test"
+		assert len(i.entities) == 0
+
+		d = { "intent": "test", "entities": { "abc": "123" } }
+		i = Interpret(d)
+		assert isinstance(i, Interpret)
+		assert i.intent == "test"
+		assert len(i.entities) == 1
+		assert i.entities["abc"] == "123"

tests/test_interpret.py

@@ -0,0 +1,99 @@
+import os, json, array, pytest, mock
+import auroraapi
+from auroraapi.globals import _config
+from auroraapi.api import APIException
+from auroraapi.audio import *
+from auroraapi.text import Text
+from auroraapi.speech import *
+
+def mock_pyaudio_record(a, b):
+	with open("tests/assets/hw.wav", "rb") as f:
+		yield array.array('h', AudioFile(f.read()).audio.raw_data)
+
+class TestSpeech(object):
+	def setup(self):
+		try:
+			_config.app_id = os.environ["APP_ID"]
+			_config.app_token = os.environ["APP_TOKEN"]
+			_config.device_id = os.environ["DEVICE_ID"]
+		except:
+			pass
+			
+	def teardown(self):
+		_config.app_id = None
+		_config.app_token = None
+		_config.device_id = None
+	
+	def test_create_no_argument(self):
+		with pytest.raises(TypeError):
+			Speech()
+	
+	def test_create_none(self):
+		with pytest.raises(TypeError):
+			Speech(None)
+	
+	def test_create_wrong_type(self):
+		with pytest.raises(TypeError):
+			Speech("string")
+	
+	def test_create(self):
+		with open("tests/assets/hw.wav", "rb") as f:
+			Speech(AudioFile(f.read()))
+
+	def test_text(self):
+		with open("tests/assets/hw.wav", "rb") as f:
+			s = Speech(AudioFile(f.read()))
+			t = s.text()
+
+			assert isinstance(t, Text)
+			assert t.text.lower().strip() == "hello world"
+
+class TestSpeechNoCreds(object):
+	def test_text(self):
+		with pytest.raises(APIException):
+			with open("tests/assets/hw.wav", "rb") as f:
+				Speech(AudioFile(f.read())).text()
+
+class TestListen(object):
+	def setup(self):
+		with open("tests/assets/hw.wav", "rb") as f:
+			self.audio_file = AudioFile(f.read())
+		try:
+			_config.app_id = os.environ["APP_ID"]
+			_config.app_token = os.environ["APP_TOKEN"]
+			_config.device_id = os.environ["DEVICE_ID"]
+		except:
+			pass
+			
+	def teardown(self):
+		_config.app_id = None
+		_config.app_token = None
+		_config.device_id = None
+
+	def test_listen(self):
+		with mock.patch('auroraapi.audio._pyaudio_record', new=mock_pyaudio_record):
+			s = listen()
+			assert isinstance(s, Speech)
+			assert isinstance(s.audio, AudioFile)
+			assert len(self.audio_file.audio) == len(s.audio.audio)
+
+	def test_continuously_listen(self):
+		with mock.patch('auroraapi.audio._pyaudio_record', new=mock_pyaudio_record):
+			for s in continuously_listen():
+				assert isinstance(s, Speech)
+				assert isinstance(s.audio, AudioFile)
+				assert len(self.audio_file.audio) == len(s.audio.audio)
+				break
+	
+	def test_listen_and_transcribe(self):
+		with mock.patch('auroraapi.audio._pyaudio_record', new=mock_pyaudio_record):
+			t = listen_and_transcribe()
+			assert isinstance(t, Text)
+			assert t.text.lower().strip() == "hello world"
+	
+	def test_continuously_listen_and_transcribe(self):
+		with mock.patch('auroraapi.audio._pyaudio_record', new=mock_pyaudio_record):
+			for t in continuously_listen_and_transcribe():
+				assert isinstance(t, Text)
+				assert t.text.lower().strip() == "hello world"
+				break