testing: add new, simplified testing infrastructure

astraw · astraw · commit d8a9e85546d3 · 2009-09-06T01:43:18.000Z
svn path=/trunk/matplotlib/; revision=7647
diff --git a/lib/matplotlib/tests/__init__.py b/lib/matplotlib/tests/__init__.py
diff --git a/lib/matplotlib/tests/test_basic.py b/lib/matplotlib/tests/test_basic.py
@@ -0,0 +1,11 @@
+from nose.tools import assert_equal
+from matplotlib.testing.decorators import knownfailureif
+
+def test_simple():
+    '''very simple example test'''
+    assert_equal(1+1,2)
+
+@knownfailureif(True)
+def test_simple_fail():
+    '''very simple example test that should fail'''
+    assert_equal(1+1,3)
diff --git a/lib/matplotlib/tests/test_transforms.py b/lib/matplotlib/tests/test_transforms.py
@@ -0,0 +1,40 @@
+from nose.tools import assert_equal
+from numpy.testing import assert_almost_equal
+from matplotlib.transforms import Affine2D
+import numpy as np
+
+def test_Affine2D_from_values():
+    points = [ [0,0],
+               [10,20],
+               [-1,0],
+               ]
+
+    t = Affine2D.from_values(1,0,0,0,0,0)
+    actual = t.transform(points)
+    expected = np.array( [[0,0],[10,0],[-1,0]] )
+    assert_almost_equal(actual,expected)
+
+    t = Affine2D.from_values(0,2,0,0,0,0)
+    actual = t.transform(points)
+    expected = np.array( [[0,0],[0,20],[0,-2]] )
+    assert_almost_equal(actual,expected)
+
+    t = Affine2D.from_values(0,0,3,0,0,0)
+    actual = t.transform(points)
+    expected = np.array( [[0,0],[60,0],[0,0]] )
+    assert_almost_equal(actual,expected)
+
+    t = Affine2D.from_values(0,0,0,4,0,0)
+    actual = t.transform(points)
+    expected = np.array( [[0,0],[0,80],[0,0]] )
+    assert_almost_equal(actual,expected)
+
+    t = Affine2D.from_values(0,0,0,0,5,0)
+    actual = t.transform(points)
+    expected = np.array( [[5,0],[5,0],[5,0]] )
+    assert_almost_equal(actual,expected)
+
+    t = Affine2D.from_values(0,0,0,0,0,6)
+    actual = t.transform(points)
+    expected = np.array( [[0,6],[0,6],[0,6]] )
+    assert_almost_equal(actual,expected)
diff --git a/setup.py b/setup.py
@@ -51,6 +51,7 @@
     'matplotlib.backends',
     'matplotlib.projections',
     'matplotlib.testing',
+    'matplotlib.tests',
 #   'matplotlib.toolkits',
     'mpl_toolkits',
     'mpl_toolkits.mplot3d',
diff --git a/test/run-mpl-test.py b/test/run-mpl-test.py
@@ -90,6 +90,9 @@
       sys.exit( 0 )
 
 ### Run nose
+args.append('.')
+args.append('matplotlib.tests.test_basic')
+args.append('matplotlib.tests.test_transforms')
 success = nose.run( argv = args,
                     plugins = [ MplNosePlugin(), KnownFailure() ] )
 
