| 1 | import unittest2 as unittest |
|---|
| 2 | import numpy as np |
|---|
| 3 | |
|---|
| 4 | from vsm.viewer.wrappers import * |
|---|
| 5 | from vsm.viewer.labeleddata import * |
|---|
| 6 | |
|---|
| 7 | |
|---|
| 8 | |
|---|
| 9 | |
|---|
| 10 | class TestViewerWrappers(unittest.TestCase): |
|---|
| 11 | |
|---|
| 12 | # TODO: Rewrite these to be independent of LDA |
|---|
| 13 | pass |
|---|
| 14 | |
|---|
| 15 | # def setUp(self): |
|---|
| 16 | |
|---|
| 17 | # from vsm.corpus.util.corpusbuilders import random_corpus |
|---|
| 18 | # from vsm.model.ldacgsseq import LdaCgsSeq |
|---|
| 19 | |
|---|
| 20 | # self.c = random_corpus(1000, 50, 0, 20, context_type='sentence', |
|---|
| 21 | # metadata=True) |
|---|
| 22 | |
|---|
| 23 | # self.m = LDAGibbs(self.c, 'sentence', K=40) |
|---|
| 24 | # self.m.train(n_iterations=50) |
|---|
| 25 | |
|---|
| 26 | |
|---|
| 27 | # def test_dist_(self): |
|---|
| 28 | |
|---|
| 29 | # li = [0,1] |
|---|
| 30 | |
|---|
| 31 | # swt = dist_word_top(self.c, self.m.word_top.T, '0') |
|---|
| 32 | # swtl = dist_word_top(self.c, self.m.word_top.T, ['0','1'], order='i') |
|---|
| 33 | # sww = dist_word_word(self.c, self.m.word_top, '0') |
|---|
| 34 | # swwl = dist_word_word(self.c, self.m.word_top, ['0','1'], order='i') |
|---|
| 35 | # std = dist_top_doc(self.c, self.m.top_doc.T, 0, 'sentence', order='i') |
|---|
| 36 | # stdl = dist_top_doc(self.c, self.m.top_doc.T, li, 'sentence') |
|---|
| 37 | # sdd = dist_doc_doc(self.c, self.m.top_doc, self.m.context_type, 0) |
|---|
| 38 | # sddl = dist_doc_doc(self.c, self.m.top_doc, self.m.context_type, li) |
|---|
| 39 | # stt = dist_top_top(self.m.word_top.T, 1) |
|---|
| 40 | # sttl = dist_top_top(self.m.word_top.T, li) |
|---|
| 41 | |
|---|
| 42 | # self.assertEqual(type(swt), LabeledColumn) |
|---|
| 43 | # self.assertEqual(type(swtl), LabeledColumn) |
|---|
| 44 | # self.assertEqual(type(sww), LabeledColumn) |
|---|
| 45 | # self.assertEqual(type(swwl), LabeledColumn) |
|---|
| 46 | # self.assertEqual(type(std), LabeledColumn) |
|---|
| 47 | # self.assertEqual(type(stdl), LabeledColumn) |
|---|
| 48 | # self.assertEqual(type(sdd), LabeledColumn) |
|---|
| 49 | # self.assertEqual(type(sddl), LabeledColumn) |
|---|
| 50 | # self.assertEqual(type(stt), LabeledColumn) |
|---|
| 51 | # self.assertEqual(type(sttl), LabeledColumn) |
|---|
| 52 | |
|---|
| 53 | |
|---|
| 54 | # def test_dismat_(self): |
|---|
| 55 | |
|---|
| 56 | # dismatw = dismat_word(['0','2','5'], self.c, self.m.word_top) |
|---|
| 57 | # dismatd = dismat_doc([0,1,2], self.c, self.m.context_type, |
|---|
| 58 | # self.m.top_doc) |
|---|
| 59 | # dismatt = dismat_top([0,1,2], self.m.word_top) |
|---|
| 60 | |
|---|
| 61 | # self.assertEqual(type(dismatw), IndexedSymmArray) |
|---|
| 62 | # self.assertEqual(type(dismatd), IndexedSymmArray) |
|---|
| 63 | # self.assertEqual(type(dismatt), IndexedSymmArray) |
|---|
| 64 | |
|---|
| 65 | |
|---|
| 66 | |
|---|
| 67 | #Define and run test suite |
|---|
| 68 | suite = unittest.TestLoader().loadTestsFromTestCase(TestViewerWrappers) |
|---|
| 69 | unittest.TextTestRunner(verbosity=2).run(suite) |
|---|
