Need help? Have a feature request? Please check out the tethne-users group .
Source code for tethne.tests.test_writers_corpus

import sys
sys.path.append('../tethne')

import unittest
import tempfile
import os
from xml.etree import ElementTree as ET
import networkx as nx
import csv

from tethne import write_documents, write_documents_dtm
from tethne.readers.wos import read

datapath = './tethne/tests/data/wos3.txt'


[docs]class WriteDocumentsTest(unittest.TestCase):
[docs]    def setUp(self):
        self.temp = tempfile.mkdtemp()
        self.target = os.path.join(self.temp, 'test')
        self.corpus = read(datapath, index_by='wosid', index_features=['citations', 'authors'])

        self.dpath = self.target + '_docs.txt'
        self.mpath = self.target + '_meta.csv'

[docs]    def test_write_documents(self):
        write_documents(self.corpus, self.target, 'citations')

        self.assertTrue(os.path.exists(self.dpath))
        self.assertTrue(os.path.exists(self.mpath))
        self.assertGreater(os.path.getsize(self.dpath), 0)
        self.assertGreater(os.path.getsize(self.mpath), 0)



[docs]class WriteDocumentsDTMTest(unittest.TestCase):
[docs]    def setUp(self):
        self.temp = tempfile.mkdtemp()
        self.target = os.path.join(self.temp, 'test')
        self.corpus = read(datapath, index_by='wosid', index_features=['citations', 'authors'])

        self.multpath = self.target + '-mult.dat'
        self.metapath = self.target + '-meta.dat'
        self.seqpath = self.target + '-seq.dat'
        self.vpath = self.target + '-vocab.dat'

[docs]    def test_write_documents_dtm(self):
        write_documents_dtm(self.corpus, self.target, 'citations')

        self.assertTrue(os.path.exists(self.multpath))
        self.assertTrue(os.path.exists(self.metapath))
        self.assertTrue(os.path.exists(self.seqpath))
        self.assertTrue(os.path.exists(self.vpath))
        self.assertGreater(os.path.getsize(self.multpath), 0)
        self.assertGreater(os.path.getsize(self.metapath), 0)
        self.assertGreater(os.path.getsize(self.seqpath), 0)
        self.assertGreater(os.path.getsize(self.vpath), 0)



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