← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/test_weave.py

  • Committer: Martin Pool
  • Date: 2005-09-13 08:04:41 UTC
  • Revision ID: mbp@sourcefrog.net-20050913080441-5a15cdaf498fdad3
- fix up weave tests for new test framework

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/selftest/test_weave.py
27
27
import testsweet
28
28
from bzrlib.weave import Weave, WeaveFormatError
29
29
from bzrlib.weavefile import write_weave, read_weave
 
30
from bzrlib.selftest import TestCase
30
31
from pprint import pformat
31
32
 
32
33
 
48
49
 
49
50
 
50
51
 
51
 
class TestBase(testsweet.TestBase):
 
52
class TestBase(TestCase):
52
53
    def check_read_write(self, k):
53
54
        """Check the weave k can be written & re-read."""
54
55
        from tempfile import TemporaryFile
111
112
        self.assertEqual(k.get(0), TEXT_0)
112
113
        self.assertEqual(k.get(1), TEXT_1)
113
114
 
114
 
        k.dump(self.TEST_LOG)
115
 
 
116
115
 
117
116
 
118
117
class InvalidAdd(TestBase):
470
469
        self.assertEqual(k.get(0),
471
470
                         ["first line"])
472
471
 
473
 
        k.dump(self.TEST_LOG)
474
 
 
475
472
 
476
473
class DivergedIncludes(TestBase):
477
474
    """Weave with two diverged texts based on version 0.
771
768
    
772
769
 
773
770
 
774
 
def testweave():
775
 
    import testsweet
776
 
    from unittest import TestSuite, TestLoader
777
 
    import testweave
778
 
 
779
 
    tl = TestLoader()
780
 
    suite = TestSuite()
781
 
    suite.addTest(tl.loadTestsFromModule(testweave))
782
 
    
783
 
    return int(not testsweet.run_suite(suite)) # for shell 0=true
784
 
 
785
 
 
786
771
if __name__ == '__main__':
787
772
    import sys
788
 
    sys.exit(testweave())
 
773
    import unittest
 
774
    sys.exit(unittest.main())
789
775