master trunk of gnowsys-studio

1 files changed, 50 insertions, 0 deletions

diff --git a/objectapp/tests/comparison.py b/objectapp/tests/comparison.py
new file mode 100644
index 0000000..7f113be
--- /dev/null
+++ b/objectapp/tests/comparison.py
@@ -0,0 +1,50 @@
+"""Test cases for Objectapp's comparison"""
+from django.test import TestCase
+
+from objectapp.models import Gbobject
+from objectapp.comparison import pearson_score
+from objectapp.comparison import VectorBuilder
+from objectapp.comparison import ClusteredModel
+
+
+class ComparisonTestCase(TestCase):
+    """Test cases for comparison tools"""
+
+    def test_pearson_score(self):
+        self.assertEquals(pearson_score([42], [42]), 0.0)
+        self.assertEquals(pearson_score([0, 1, 2], [0, 1, 2]), 0.0)
+        self.assertEquals(pearson_score([0, 1, 3], [0, 1, 2]),
+                          0.051316701949486232)
+        self.assertEquals(pearson_score([0, 1, 2], [0, 1, 3]),
+                          0.051316701949486232)
+
+    def test_clustered_model(self):
+        params = {'title': 'My gbobject 1', 'content': 'My content 1',
+                  'tags': 'objectapp, test', 'slug': 'my-gbobject-1'}
+        Gbobject.objects.create(**params)
+        params = {'title': 'My gbobject 2', 'content': 'My content 2',
+                  'tags': 'objectapp, test', 'slug': 'my-gbobject-2'}
+        Gbobject.objects.create(**params)
+        cm = ClusteredModel(Gbobject.objects.all())
+        self.assertEquals(cm.dataset().values(), ['1', '2'])
+        cm = ClusteredModel(Gbobject.objects.all(),
+                            ['title', 'excerpt', 'content'])
+        self.assertEquals(cm.dataset().values(), ['My gbobject 1  My content 1',
+                                                  'My gbobject 2  My content 2'])
+
+    def test_vector_builder(self):
+        vectors = VectorBuilder(Gbobject.objects.all(),
+                                ['title', 'excerpt', 'content'])
+        params = {'title': 'My gbobject 1', 'content':
+                  'This is my first content',
+                  'tags': 'objectapp, test', 'slug': 'my-gbobject-1'}
+        Gbobject.objects.create(**params)
+        params = {'title': 'My gbobject 2', 'content':
+                  'My second gbobject',
+                  'tags': 'objectapp, test', 'slug': 'my-gbobject-2'}
+        Gbobject.objects.create(**params)
+        columns, dataset = vectors()
+        self.assertEquals(columns, ['content', 'This', 'my', 'is', '1',
+                                    'second', '2', 'first'])
+        self.assertEquals(dataset.values(), [[1, 1, 1, 1, 1, 0, 0, 1],
+                                             [0, 0, 0, 0, 0, 1, 1, 0]])