add test for txt cbeta

cltk · Mar 5, 2016 · e89e7b6 · e89e7b6
1 parent 760aca1
commit e89e7b6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/cltk/tests/test_corpus.py b/cltk/tests/test_corpus.py
@@ -303,6 +303,15 @@ def test_git_import_tib_lexica_tdc(self):
         file_exists = os.path.isfile(file)
         self.assertTrue(file_exists)
 
+    def test_git_import_chinese_cbeta_txt(self):
+        """Test import of plaintext CBETA."""
+        corpus_importer = CorpusImporter('chinese')
+        corpus_importer.import_corpus('chinese_text_cbeta_txt')
+        file_rel = os.path.join('~/cltk_data/chinese/text/chinese_text_cbeta_txt/README.md')
+        file = os.path.expanduser(file_rel)
+        file_exists = os.path.isfile(file)
+        self.assertTrue(file_exists)
+
     def test_show_corpora_bad_lang(self):
         """Test failure of importer upon selecting unsupported language."""
         with self.assertRaises(AssertionError):