/
test_sentence_classifier.py
51 lines (38 loc) · 1.7 KB
/
test_sentence_classifier.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
# -*- coding: utf-8 -*-
import unittest
from sentence import Sentence
from sentence_classifier import SentenceClassifier
import pdb
class TestSentenceClassifier(unittest.TestCase):
def setUp(self):
sentence_1 = Sentence('今すぐ泳ぎなさい', '')
self.sc_1 = SentenceClassifier(sentence_1)
sentence_2 = Sentence('魚雷を発射するとよい', '')
self.sc_2 = SentenceClassifier(sentence_2)
sentence_3 = Sentence('島風を育ててはいかがでしょう', '')
self.sc_3 = SentenceClassifier(sentence_3)
sentence_4 = Sentence('はい、大丈夫です!', '')
self.sc_4 = SentenceClassifier(sentence_4)
sentence_5 = Sentence('浴室は使用前に一度、熱湯消毒をしましょう!', '')
self.sc_5 = SentenceClassifier(sentence_5)
def test_cleared_one_hurdle_problem_r_i(self):
result = self.sc_1.cleared_one_hurdle_problem_r_i()
self.assertEqual(result, 0)
def test_cleared_two_hurdle_problem_r_i(self):
result = self.sc_2.cleared_two_hurdle_problem_r_i()
self.assertEqual(result, 1)
result_2 = self.sc_3.cleared_two_hurdle_problem_r_i()
self.assertEqual(result_2, 4)
def test_direction_r_i(self):
result_1 = self.sc_1.direction_r_i()
self.assertEqual(result_1, 0)
result_2 = self.sc_2.direction_r_i()
self.assertEqual(result_2, 1)
result_3 = self.sc_3.direction_r_i()
self.assertEqual(result_3, 4)
result_4 = self.sc_4.direction_r_i()
self.assertEqual(result_4, False)
result_5 = self.sc_5.direction_r_i()
self.assertEqual(result_5, 2)
if __name__ == '__main__':
unittest.main()