test/compat_test.py

   1 #!/usr/bin/env python
   2 # encoding: utf-8
   3
   4 from __future__ import absolute_import, division, unicode_literals
   5
   6 import os
   7 import unittest
   8
   9 from cola import compat
  10
  11
  12 class CompatTestCase(unittest.TestCase):
  13     """Tests the compat module"""
  14
  15     def test_setenv(self):
  16         """Test the core.decode function
  17         """
  18         key = 'COLA_UNICODE_TEST'
  19         value = '字龍'
  20         compat.setenv(key, value)
  21         self.assertTrue(key in os.environ)
  22         self.assertTrue(os.getenv(key))
  23
  24         compat.unsetenv(key)
  25         self.assertFalse(key in os.environ)
  26         self.assertFalse(os.getenv(key))
  27
  28
  29 if __name__ == '__main__':
  30     unittest.main()