#!/usr/bin/python
import unittest
import ps2

# For these tests to work, you must have a file named ps2.py
# which contains a function longest_substring(s, t).

class TestPS2(unittest.TestCase):
    def setUp(self):
        pass

    def test1(self):
        ans = ps2.longest_substring('','')
        self.assertEqual(ans, '')

    def test2(self):
        ans = ps2.longest_substring('aaa','bbb')
        self.assertEqual(ans, '')

    def test3(self):
        ans = ps2.longest_substring('aaa','aaa')
        self.assertEqual(ans, 'aaa')

    def test4(self):
        ans = ps2.longest_substring('the quick brown fox','THE QUICK BROWN FOX')
        self.assertEqual(ans, ' ')

    def test5(self):
        s_file = open('/mit/6.006/dna/chr2_first_10000')
        t_file = open('/mit/6.006/dna/chr2a_first_10000')
        s = s_file.readline()
        t = t_file.readline()
        ans = ps2.longest_substring(s, t)
        self.assertEqual(ans, 'gaattttacagccatatggttagagcaaactctattcaaattaaataaaattactcaggatgtggagtatcccaggacagaatacatcatgtgaaaaagcatttaTGCTACAAATTACTATGGTAATTATGCTACAAATTTATGGTACCATAAATTACCATAGTAATTTGTAGCATAAATTTGTACTATGGTACAAATTACATGGGAGAGTGAAGGTGGGTTAAAACATTCATATTAAAGAACTTCCACTCAGATTGCAAGAAAAGAGAGAGGAATGGAGATGGTAGCACAAGTCCCTACAATAAAAGTAGATGTTTTGAGATCAGTTCTATTTGTTCTGACAAAAATTAAAGACAGAAACCAAAGTTTAGCCTGAGGCAACAATTAGTTGGG')

if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(TestPS2)
    unittest.TextTestRunner(verbosity=2).run(suite)