def test_classify_predict(self):
self.test_data_loader = DataLoader()
self.test_japanese_wiki_data = 'test/test_data/jawiki_test.txt'
test_word2index, test_index2word, test_window_data, \
test_X_ik, test_weightinhg_dict = self.test_data_loader.load_data(file_name=self.test_japanese_wiki_data) # noqa
model_name = '../models/glove_wiki/glove_model_40.pth'
output_file = 'test/test_data/glove_classify_model.pkl'
classifier = Classifier(model_name=model_name)
print(test_word2index)
classes = classifier.classify_predict(word='の',
classify_model_name=output_file,
word2index=test_word2index)
assert 2 == classes
classes = classifier.classify_predict(word='どうよ?',
classify_model_name=output_file,
word2index=test_word2index)
assert 9999 == classes
