31 lines
774 B
Python
31 lines
774 B
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
#
|
||
|
|
# Copyright (C) 2010 Radim Rehurek <radimrehurek@seznam.cz>
|
||
|
|
# Licensed under the GNU LGPL v2.1 - http://www.gnu.org/licenses/lgpl.html
|
||
|
|
|
||
|
|
"""
|
||
|
|
Automated tests for checking transformation algorithms (the models package).
|
||
|
|
"""
|
||
|
|
|
||
|
|
import logging
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from gensim.topic_coherence import aggregation
|
||
|
|
|
||
|
|
|
||
|
|
class TestAggregation(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.confirmed_measures = [1.1, 2.2, 3.3, 4.4]
|
||
|
|
|
||
|
|
def testArithmeticMean(self):
|
||
|
|
"""Test arithmetic_mean()"""
|
||
|
|
obtained = aggregation.arithmetic_mean(self.confirmed_measures)
|
||
|
|
expected = 2.75
|
||
|
|
self.assertEqual(obtained, expected)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
logging.root.setLevel(logging.WARNING)
|
||
|
|
unittest.main()
|