/[thuban]/branches/WIP-pyshapelib-bramz/test/support.py

Diff of /branches/WIP-pyshapelib-bramz/test/support.py

Parent Directory | Revision Log | View Patch Patch

-revision 1263 by jonathan,
Fri Jun 20 14:15:43 2003 UTC
+revision 1605 by bh,
Tue Aug 19 11:00:40 2003 UTC
 Line 15 
 __version__ = "$Revision$"
  import os, sys
  import unittest
+ import traceback
+ import postgissupport
  def thuban_dir():
      """Return the directory containing the Thuban package"""
-Line 48 
 def initthuban():
+Line 52 
 def initthuban():
          import thubaninit
          _initthuban_done = 1
- def run_tests():
-     """Frontend for unittest.main that prints some additional debug information
-     After calling unittest.main, run the garbage collector and print
+ #
-     uncollected objects. Also print any un-unsubscribed messages.
+ # Special test runner and result that support skipping tests
+ #
+ class SkipTest(Exception):
+     """Exception to raise in tests that are skipped for some reason
+     For instance, since gdal support is optional, test cases that
+     require gdal raise this exception to indicate that they are skipped.
+     Skipped is different from failure or error in that it is expected
+     under certain circumstances.
+     """
+ class ThubanTestResult(unittest._TextTestResult):
+     def __init__(self, stream, descriptions, verbosity):
+         unittest._TextTestResult.__init__(self, stream, descriptions,
+                                           verbosity)
+         self.skipped_tests = {}
+     def add_skipped_test(self, test, exc):
+         reason = str(exc)
+         self.skipped_tests.setdefault(reason, []).append(test)
+     def count_skipped(self):
+         sum = 0
+         for tests in self.skipped_tests.values():
+             sum += len(tests)
+         return sum
+     def addError(self, test, err):
+         """Extend inherited method to handle SkipTest exceptions specially
+         """
+         #print "addError", test, err
+         if isinstance(err[1], SkipTest):
+             self.add_skipped_test(test, err[1])
+             if self.showAll:
+                 self.stream.writeln("skipped")
+             elif self.dots:
+                 self.stream.write('S')
+         else:
+             unittest._TextTestResult.addError(self, test, err)
+     def printErrors(self):
+         if self.skipped_tests:
+             if self.dots or self.showAll:
+                 self.stream.writeln()
+             self.stream.writeln("Skipped tests:")
+             for reason, tests in self.skipped_tests.items():
+                 self.stream.writeln("  %s:" % reason)
+                 for test in tests:
+                     self.stream.writeln("    " + test.id())
+         unittest._TextTestResult.printErrors(self)
+ class ThubanTestRunner(unittest.TextTestRunner):
+     def _makeResult(self):
+         return ThubanTestResult(self.stream, self.descriptions, self.verbosity)
+     def run(self, test):
+         result = unittest.TextTestRunner.run(self, test)
+         self.stream.writeln("skipped = %d" % result.count_skipped())
+         return result
+ class ThubanTestProgram(unittest.TestProgram):
+     def runTests(self):
+         """Extend inherited method so that we use a ThubanTestRunner"""
+         print "ThubanTestProgram.runTests"
+         self.testRunner = ThubanTestRunner(verbosity = self.verbosity)
+         unittest.TestProgram.runTests(self)
+ def execute_as_testsuite(callable, *args, **kw):
+     """Call callable  with args as if it were the entry point to the test suite
+     Return watever callable returns.
+     This is a helper function for run_tests and runtests.py. Call
+     callable in a try-finally block and run some cleanup and print some
+     additional information in the finally block.
+     The additionaly information include:
+      - A list of uncollected objects (after an explicit garbage
+        collector call)
+      - any unsubscribed messages
      """
      try:
-         unittest.main()
+         return callable(*args, **kw)
      finally:
          # This has to be in a finally clause because unittest.main()
          # ends with a sys.exit to make sure that the process exits with
          # an appropriate exit code
+         # Shutdown the postgis server if it's running
+         try:
+             postgissupport.shutdown_test_server()
+         except:
+             traceback.print_exc()
+         # Print additional information
          print_additional_summary()
+ def run_tests():
+     """Frontend for unittest.main that prints some additional debug information
+     After calling unittest.main, run the garbage collector and print
+     uncollected objects. Also print any un-unsubscribed messages.
+     """
+     execute_as_testsuite(ThubanTestProgram)
  def print_additional_summary():
      """Print some additional summary information after tests have been run"""
      print_garbage_information()
-Line 82 
 def print_garbage_information():
+Line 189 
 def print_garbage_information():
          print gc.garbage
      Thuban.Lib.connector._the_connector.print_connections()
+ #
  def create_temp_dir():
      """Create a temporary directory and return its name.
-Line 160 
 class FileLoadTestCase(unittest.TestCase
+Line 268 
 class FileLoadTestCase(unittest.TestCase
          file.close()
- class FloatComparisonMixin(unittest.TestCase):
+ class FloatComparisonMixin:
      """
      Mixin class for tests comparing floating point numbers.
-Line 171 
 class FloatComparisonMixin(unittest.Test
+Line 279 
 class FloatComparisonMixin(unittest.Test
      fp_epsilon = 1e-6
      fp_inf = float('1e1000')   # FIXME: hack for infinite
-     def assertFloatEqual(self, test, value):
+     def assertFloatEqual(self, test, value, epsilon = None):
          """Assert equality of test and value with some tolerance.
          Assert that the absolute difference between test and value is
          less than self.fp_epsilon.
          """
+         if epsilon is None:
+             epsilon = self.fp_epsilon
          if abs(test) == self.fp_inf:
              self.assertEqual(test, value)
          else:
-             self.assert_(self.fp_epsilon > abs(test - value),
+             self.assert_(epsilon > abs(test - value),
-                      "abs(%g - %g) >= %g" % (test, value, self.fp_epsilon))
+                      "abs(%g - %g) >= %g" % (test, value, epsilon))
      def assertFloatSeqEqual(self, test, value, epsilon = None):
          """Assert equality of the sequences test and value with some tolerance.
-Line 191 
 class FloatComparisonMixin(unittest.Test
+Line 301 
 class FloatComparisonMixin(unittest.Test
          value in test and value is less than the optional parameter
          epsilon. If epsilon is not given use self.fp_epsilon.
          """
-         if epsilon is None:
-             epsilon = self.fp_epsilon
          for i in range(len(test)):
-             self.assert_(epsilon > abs(test[i] - value[i]),
+             self.assertFloatEqual(test[i], value[i], epsilon)
-                          "abs(%g - %g) >= %g" % (test[i], value[i], epsilon))
+     def assertPointListEquals(self, test, value):
+         """Assert equality of two lists of lists of tuples of float
+         This assertion is usually used to compare the geometry of shapes
+         as returned by a Shape object's Points() method, hence the name.
+         """
+         for i in range(len(test)):
+             self.assertEquals(len(test[i]), len(value[i]))
+             for j in range(len(test[i])):
+                 self.assertFloatSeqEqual(test[i][j], value[i][j])
  class SubscriberMixin:

 Legend:



Removed from v.1263
 


changed lines


 
Added in v.1605
 Legend:



Removed from v.1263
 


changed lines


 
Added in v.1605
-Removed from v.1263
+Added in v.1605

[email protected]	ViewVC Help
Powered by ViewVC 1.1.26