my_tests.py

"""my unit-testing file"""
import unittest
import final
import graph
import functools
import itertools


class MyTests(unittest.TestCase):

    # def test_tokenization1(self):
    #     temp = None
    #     self.assertTrue(final.Tokenization.is_proper_enum(temp))
    #
    # def test_tokenization2(self):
    #     temp = final.Tokenization.word
    #     self.assertTrue(final.Tokenization.is_proper_enum(temp))
    #
    # def test_constructor(self):
    #     rw = final.RandomWriter(1, final.Tokenization.word)
    #
    # def test_graph(self):
    #     g = graph.Graph()
    #     g.add_vertex(1)
    #     g.add_vertex(2)
    #     print(g.vertices)
    #     for v in g.vertices:
    #         if v == 1:
    #             temp = g[v]
    #             temp.add_edge(g[2], '3')
    #             print(temp)
    #             print(temp.get_edge('3').dest_vertex)
    #
    # def test_bytes(self):
    #     print("TESTING BYTES:\n")
    #     rw = final.RandomWriter(1, final.Tokenization.byte)
    #     rw.train_iterable(b"\xfe\xff\x02\xfe\x03\xfe")
    #     print(rw.chain.vertices)
    #     rw.chain.print_graph()
    #     for x in itertools.islice(rw.generate(), 100):
    #         print(x)
    #
    # def test_words(self):
    #     print("TESTING WORDS:\n")
    #     rw = final.RandomWriter(1, final.Tokenization.word)
    #     rw.train_iterable("a the word the")
    #     print(rw.chain.vertices)
    #     rw.chain.print_graph()

    def test_url_char(self):
        print("TESTING WORD URL:\n")
        rw = final.RandomWriter(3, final.Tokenization.character)
        rw.train_url("http://www.singingwizard.org/stuff/pg24132.txt")
        print(rw.chain.vertices)
        rw.chain.print_graph()
        for x in itertools.islice(rw.generate(), 100):
            print(x)

    # def test_none(self):
    #     print("TESTING NONE:\n")
    #     rw = final.RandomWriter(2, final.Tokenization.none)
    #     rw.train_iterable((1,2,3,4,5,5,4,3,2,1))
    #     print(rw.chain.vertices)
    #     rw.chain.print_graph()
    #     rw.generate90


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