#!/usr/bin/env python
"""Test coordinate classes."""

import sys

try:
    import unittest2 as unittest  # Python 2.6
except ImportError:
    import unittest

import heatmap as hm


class Tests(unittest.TestCase):

    # To remove Python 3's
    # "DeprecationWarning: Please use assertRaisesRegex instead"
    if sys.version_info[0] == 2:
        assertRaisesRegex = unittest.TestCase.assertRaisesRegexp

    def test_basic(self):
        '''Test Configuration class.'''

        # Act
        config = hm.Configuration(use_defaults=True)

        # Assert
        self.assertEqual(config.margin, 0)
        self.assertEqual(config.frequency, 1)

    def test_fill_missing_no_input(self):
        '''Test Configuration class.'''
        # Arrange
        config = hm.Configuration(use_defaults=True)

        # Act / Assert
        with self.assertRaisesRegex(ValueError, "no input specified"):
            config.fill_missing()


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

@@ -152,16 +152,114 @@
 ittest%0A%0A
+ROOT_DIR = os.path.split(os.path.abspath(os.path.dirname(__file__)))%5B0%5D%0Asys.path.append(ROOT_DIR)%0A
 import h
@@ -563,17 +563,16 @@
 ass.'''%0A
-%0A
         

Update sys.path to import heatmap