import unittest
from performance import SLUGS,cue,plan


class PerformanceTests(unittest.TestCase):
    def test_all_forty_canonical_labels(self):
        self.assertEqual(len(SLUGS),40)
        for slug in SLUGS:self.assertEqual(cue(slug)['emotion'],slug)

    def test_no_emotion_inference_from_spoken_words(self):
        p=plan({'general':'calm','tagged':'[2.0 seconds duration] (angry, intense) I am not angry.'})
        self.assertEqual(p['sentences'][0]['emotion'],'anger')

    def test_negated_direction(self):self.assertNotEqual(cue('not angry, relieved')['emotion'],'anger')

    def test_timed_sentence_changes(self):
        p=plan({'tagged':'[0.3 seconds pause] [2.0 seconds duration] (sad) Oh no. [0.4 seconds pause] [1.6 seconds duration] (relieved) We are safe.'})
        a,b=p['sentences'];self.assertEqual(a['emotion'],'sadness');self.assertEqual(b['emotion'],'relief')
        self.assertAlmostEqual(a['start_s'],.3);self.assertAlmostEqual(a['end_s'],2.3)
        self.assertAlmostEqual(b['start_s'],2.7);self.assertAlmostEqual(b['end_s'],4.3)

    def test_burst_not_spoken_sentence(self):
        p=plan({'tagged':'(sad) [1.2 seconds duration] Oh no. (wistful sigh, 0.3 seconds)'})
        self.assertEqual(len(p['sentences']),1);self.assertEqual(p['sentences'][0]['emotion'],'sadness')

    def test_empty_and_mix(self):
        self.assertTrue(plan({})['sentences']);c=cue('strong sadness with relief')
        self.assertEqual(c['secondary'],'relief');self.assertGreater(c['intensity'],.8)

if __name__=='__main__':unittest.main()