/usr/lib/python2.7/dist-packages/ginga/tests/test_cmap.py is in python-ginga 2.6.1-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 | #
# Unit Tests for the cmap.py functions
#
# Rajul Srivastava (rajul09@gmail.com)
#
import unittest
import logging
import numpy as np
import ginga.cmap
from ginga.cmap import ColorMap
class TestError(Exception):
pass
class TestCmap(unittest.TestCase):
def setUp(self):
pass
def test_ColorMap_init(self):
test_clst = tuple([(x, x, x)
for x in np.linspace(0, 1, ginga.cmap.min_cmap_len)])
test_color_map = ColorMap('test-name', test_clst)
expected = 'test-name'
actual = test_color_map.name
assert expected == actual
expected = ginga.cmap.min_cmap_len
actual = len(test_color_map.clst)
assert expected == actual
expected = (0.0, 0.0, 0.0)
actual = test_color_map.clst[0]
assert np.allclose(expected, actual)
expected = (1.0, 1.0, 1.0)
actual = test_color_map.clst[-1]
assert np.allclose(expected, actual)
def test_ColorMap_init_exception(self):
self.assertRaises(TypeError, ColorMap, 'test-name')
def test_cmaps(self):
count = 0
for attribute_name in dir(ginga.cmap):
if attribute_name.startswith('cmap_'):
count = count + 1
expected = count
actual = len(ginga.cmap.cmaps)
assert expected == actual
def test_add_cmap(self):
test_clst = tuple([(x, x, x)
for x in np.linspace(0, 1, ginga.cmap.min_cmap_len)])
ginga.cmap.add_cmap('test-name', test_clst)
expected = ColorMap('test-name', test_clst)
actual = ginga.cmap.cmaps['test-name']
assert expected.name == actual.name
assert expected.clst == actual.clst
# Teardown
del ginga.cmap.cmaps['test-name']
def test_add_cmap_exception(self):
test_clst = ((0.0, 0.0, 0.0), (1.0, 1.0, 1.0))
self.assertRaises(
AssertionError, ginga.cmap.add_cmap, 'test-name', test_clst)
def test_get_cmap(self):
test_clst = tuple([(x, x, x)
for x in np.linspace(0, 1, ginga.cmap.min_cmap_len)])
ginga.cmap.add_cmap('test-name', test_clst)
expected = ColorMap('test-name', test_clst)
actual = ginga.cmap.get_cmap('test-name')
assert expected.name == actual.name
assert expected.clst == actual.clst
# Teardown
del ginga.cmap.cmaps['test-name']
def test_get_cmap_exception(self):
self.assertRaises(KeyError, ginga.cmap.get_cmap, 'non-existent-name')
def test_get_names(self):
names = []
for attribute_name in dir(ginga.cmap):
if attribute_name.startswith('cmap_'):
names.append(attribute_name[5:])
expected = sorted(names, key=lambda s: s.lower())
actual = ginga.cmap.get_names()
assert expected == actual
# TODO: Add tests for matplotlib functions
def tearDown(self):
pass
if __name__ == '__main__':
unittest.main()
# END
|