1. 1
2. 2
3. 3
4. 4
5. 5
6. 6
7. 7
8. 8
9. 9
10. 10
11. 11
12. 12
13. 13
14. 14
15. 15
16. 16

import unittest
from pingshui_rhyme import PingZeClassifier

class TestPingZeClassifier(unittest.TestCase):

    def setUp(self):
        self.classifier = PingZeClassifier()

    def test_classify(self):
        sentence = "知否？知否？應是綠肥紅瘦。"
        result = self.classifier.classify(sentence)
        expected = ['ping', 'ze', 'unknown', 'ping', 'ze', 'unknown', 'ping', 'ze', 'ze', 'ping', 'ping', 'ze', 'unknown']
        self.assertEqual(result, expected)

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