| import unittest |
| import numpy as np |
| from identity import IdentityManager |
| from audio_sync import estimate_echo_delay |
|
|
|
|
| class IdentityTests(unittest.TestCase): |
| def setUp(self): |
| self.a=np.eye(192,dtype=np.float32)[0];self.b=np.eye(192,dtype=np.float32)[1] |
| self.r=np.eye(192,dtype=np.float32)[2];self.now=100 |
| self.manager=IdentityManager(robo=self.r,clock=lambda:self.now) |
|
|
| def say(self,vector,text): |
| for _ in range(4):self.manager.observe({'embedding':vector,'kind':'speech','confidence':.9}) |
| return self.manager.finish(text,enabled=True) |
|
|
| def test_multiuser_and_unknown(self): |
| self.assertFalse(self.say(self.a,'How are you?')['accepted']) |
| self.assertTrue(self.say(self.a,'Hey Robo, I would like to talk to you.')['new_user']) |
| self.assertTrue(self.say(self.a,'Tell me something.')['accepted']) |
| self.assertTrue(self.say(self.b,'Hallo Robo, ich bin auch hier.')['new_user']) |
| self.assertEqual(len(self.manager.users),2) |
| self.assertFalse(self.say(self.r,'Hey Robo, this is your own voice.')['accepted']) |
|
|
| def test_uncertainty_expiry_and_clear(self): |
| self.manager.observe({'embedding':None,'kind':'overlap','confidence':.9}) |
| self.assertFalse(self.manager.finish('Hey Robo',enabled=True)['accepted']) |
| self.say(self.a,'Hey Robo, please enroll me.');self.now+=301 |
| self.assertFalse(self.say(self.a,'Are you there?')['accepted']) |
| self.manager.clear();self.assertEqual(self.manager.status()['users'],[]) |
|
|
| def test_calibration(self): |
| rng=np.random.default_rng(7);reference=rng.normal(0,.03,32000) |
| for ms in [0,13.5,80,180,440]: |
| delay=round(ms*16);mic=np.r_[np.zeros(delay),reference][:len(reference)]*.1 |
| result=estimate_echo_delay(mic,reference) |
| self.assertTrue(result['available']);self.assertAlmostEqual(result['delay_ms'],ms,places=1) |
| self.assertFalse(estimate_echo_delay(np.zeros_like(reference),reference)['available']) |
| self.assertFalse(estimate_echo_delay(rng.normal(size=len(reference)),reference)['available']) |
|
|
|
|
| if __name__=='__main__':unittest.main() |
|
|
