test_grc_features.py

# pylint: disable = missing-function-docstring, missing-class-docstring
'''Tests'''
import unittest
from functools import reduce

from greek_features import * # pylint: disable = wildcard-import, unused-wildcard-import

class TestGreekFeatures(unittest.TestCase):

	def setUp(self):
		pass

	def test_freq_indefinite_pronoun_in_non_interrogative_sentence1(self):
		file = 'test test. test test test? test test test; test test. test.'
		result = freq_indefinite_pronoun_in_non_interrogative_sentence(text=file)
		expected = 0
		self.assertEqual(expected, result)

	def test_freq_indefinite_pronoun_in_non_interrogative_sentence2(self):
		file = 'τις τινός τινός, αβγ δεζ. ηθικλ τις μνξοπρσ; τινός. τω τι τυφχψω.'
		result = freq_indefinite_pronoun_in_non_interrogative_sentence(text=file)
		expected = 5 / 39
		self.assertEqual(expected, result)

	def test_freq_sentences_with_vocative_omega(self):
		file = 'ὦ α βγδ ὦ εζθικλ. μνξ ὦ οπ, ρσὦ τυ; φχψω. ὦ.'
		result = freq_sentences_with_vocative_omega(text=file)
		expected = 3 / 4
		self.assertEqual(expected, result)

	def test_freq_sentence_with_relative_clause(self):
		file = 'ὅς αφνιοςε ςοεφπκ. ὃ ςενςεο; ΝΕΣΟΝΦΕΙΝΟΣΙ ὧν. ὧν. νψςειοιοςνφψλε.'
		result = freq_sentence_with_relative_clause(text=file)
		expected = 4 / 5
		self.assertEqual(expected, result)

	def test_mean_sentence_length1(self):
		file = 'ὅς αφνιοςε ςοεφπκ. ὃ ςενςεο; ΝΕΣΟΝΦΕΙΝΟΣΙ ὧν. ὧν. νψςειοιοςνφψλε.'
		result = mean_sentence_length(text=file)
		expected = (16 + 8 + 15 + 3 + 15) / 5
		self.assertEqual(expected, result)

	def test_variance_of_sentence_length1(self):
		file = 'ὅς αφνιοςε ςοεφπκ. ὃ ςενςεο; ΝΕΣΟΝΦΕΙΝΟΣΙ ὧν. ὧν. νψςειοιοςνφψλε.'
		result = variance_of_sentence_length(text=file)
		vals = (16, 8, 15, 3, 15)
		mean = sum(vals) / len(vals)
		expected = reduce(lambda cur_sum, val: cur_sum + (val - mean) ** 2, vals, 0) / len(vals)
		self.assertEqual(expected, result)

	def test_freq_interrogatives1(self):
		#Includes GREEK QUESTION MARK after ofiihwio;
		file = 'a b ccccccc. aaafew aaedwp bbdinwe; bnoirenfiob; ads ofiihwio; freino. daieof; frinoe.'
		result = freq_interrogatives(text=file)
		expected = 4 / 7
		self.assertEqual(expected, result)

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