Skip to content

Commit

Permalink
Strict PEP8 compliance for printed arrays
Browse files Browse the repository at this point in the history
  • Loading branch information
@joferkington
joferkington committed Sep 27, 2014
1 parent abf8b87 commit 689b3e1
Showing 1 changed file with 71 additions and 70 deletions.
141 changes: 71 additions & 70 deletions lib/matplotlib/tests/test_colors.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -254,6 +254,7 @@ def test_light_source_topo_surface():
ax.imshow(rgb)
ax.set(xticks=[], yticks=[])


def test_light_source_shading_default():
"""Array comparison test for the default "hsv" blend mode. Ensure the
default result doesn't change without warning."""
Expand All @@ -264,41 +265,41 @@ def test_light_source_shading_default():
ls = mcolors.LightSource(315, 45)
rgb = ls.shade(z, cmap)

r = np.array([[[ 0.87, 0.85, 0.9 , 0.9 , 0.82, 0.62, 0.34, 0. ],
[ 0.85, 0.94, 0.99, 1. , 1. , 0.96, 0.62, 0.17],
[ 0.9 , 0.99, 1. , 1. , 1. , 1. , 0.71, 0.33],
[ 0.9 , 1. , 1. , 1. , 1. , 0.98, 0.51, 0.29],
[ 0.82, 1. , 1. , 1. , 1. , 0.64, 0.25, 0.13],
[ 0.62, 0.96, 1. , 0.98, 0.64, 0.22, 0.06, 0.03],
[ 0.34, 0.62, 0.71, 0.51, 0.25, 0.06, 0. , 0.01],
[ 0. , 0.17, 0.33, 0.29, 0.13, 0.03, 0.01, 0. ]],

[[ 0.87, 0.79, 0.83, 0.8 , 0.66, 0.44, 0.23, 0. ],
[ 0.79, 0.88, 0.93, 0.92, 0.83, 0.66, 0.38, 0.1 ],
[ 0.83, 0.93, 0.99, 1. , 0.92, 0.75, 0.4 , 0.18],
[ 0.8 , 0.92, 1. , 0.99, 0.93, 0.75, 0.28, 0.14],
[ 0.66, 0.83, 0.92, 0.93, 0.87, 0.44, 0.12, 0.06],
[ 0.44, 0.66, 0.75, 0.75, 0.44, 0.12, 0.03, 0.01],
[ 0.23, 0.38, 0.4 , 0.28, 0.12, 0.03, 0. , 0. ],
[ 0. , 0.1 , 0.18, 0.14, 0.06, 0.01, 0. , 0. ]],

[[ 0.87, 0.75, 0.78, 0.73, 0.55, 0.33, 0.16, 0. ],
[ 0.75, 0.85, 0.9 , 0.86, 0.71, 0.48, 0.23, 0.05],
[ 0.78, 0.9 , 0.98, 1. , 0.82, 0.51, 0.21, 0.08],
[ 0.73, 0.86, 1. , 0.97, 0.84, 0.47, 0.11, 0.05],
[ 0.55, 0.71, 0.82, 0.84, 0.71, 0.2 , 0.03, 0.01],
[ 0.33, 0.48, 0.51, 0.47, 0.2 , 0.02, 0. , 0. ],
[ 0.16, 0.23, 0.21, 0.11, 0.03, 0. , 0. , 0. ],
[ 0. , 0.05, 0.08, 0.05, 0.01, 0. , 0. , 0. ]],

[[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ]]])
r = np.array([[[0.87, 0.85, 0.90, 0.90, 0.82, 0.62, 0.34, 0.00],
[0.85, 0.94, 0.99, 1.00, 1.00, 0.96, 0.62, 0.17],
[0.90, 0.99, 1.00, 1.00, 1.00, 1.00, 0.71, 0.33],
[0.90, 1.00, 1.00, 1.00, 1.00, 0.98, 0.51, 0.29],
[0.82, 1.00, 1.00, 1.00, 1.00, 0.64, 0.25, 0.13],
[0.62, 0.96, 1.00, 0.98, 0.64, 0.22, 0.06, 0.03],
[0.34, 0.62, 0.71, 0.51, 0.25, 0.06, 0.00, 0.01],
[0.00, 0.17, 0.33, 0.29, 0.13, 0.03, 0.01, 0.00]],

[[0.87, 0.79, 0.83, 0.80, 0.66, 0.44, 0.23, 0.00],
[0.79, 0.88, 0.93, 0.92, 0.83, 0.66, 0.38, 0.10],
[0.83, 0.93, 0.99, 1.00, 0.92, 0.75, 0.40, 0.18],
[0.80, 0.92, 1.00, 0.99, 0.93, 0.75, 0.28, 0.14],
[0.66, 0.83, 0.92, 0.93, 0.87, 0.44, 0.12, 0.06],
[0.44, 0.66, 0.75, 0.75, 0.44, 0.12, 0.03, 0.01],
[0.23, 0.38, 0.40, 0.28, 0.12, 0.03, 0.00, 0.00],
[0.00, 0.10, 0.18, 0.14, 0.06, 0.01, 0.00, 0.00]],

[[0.87, 0.75, 0.78, 0.73, 0.55, 0.33, 0.16, 0.00],
[0.75, 0.85, 0.90, 0.86, 0.71, 0.48, 0.23, 0.05],
[0.78, 0.90, 0.98, 1.00, 0.82, 0.51, 0.21, 0.08],
[0.73, 0.86, 1.00, 0.97, 0.84, 0.47, 0.11, 0.05],
[0.55, 0.71, 0.82, 0.84, 0.71, 0.20, 0.03, 0.01],
[0.33, 0.48, 0.51, 0.47, 0.20, 0.02, 0.00, 0.00],
[0.16, 0.23, 0.21, 0.11, 0.03, 0.00, 0.00, 0.00],
[0.00, 0.05, 0.08, 0.05, 0.01, 0.00, 0.00, 0.00]],

[[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00]]])
# Result stored transposed and rounded for for more compact display...
assert_array_almost_equal(rgb, r.T, decimal=2)

Expand All @@ -315,41 +316,41 @@ def test_light_source_masked_shading():
ls = mcolors.LightSource(315, 45)
rgb = ls.shade(z, cmap)

r = np.array([[[ 1. , 0.95, 0.96, 0.94, 0.86, 0.67, 0.4 , 0.03],
[ 0.95, 0.99, 1. , 1. , 1. , 0.98, 0.67, 0.19],
[ 0.96, 1. , 1. , 1. , 1. , 1. , 0.78, 0.36],
[ 0.94, 1. , 1. , 0. , 0. , 1. , 0.55, 0.32],
[ 0.86, 1. , 1. , 0. , 0. , 1. , 0.27, 0.14],
[ 0.67, 0.98, 1. , 1. , 1. , 1. , 0.07, 0.03],
[ 0.4 , 0.67, 0.78, 0.55, 0.27, 0.07, 0. , 0.01],
[ 0.03, 0.19, 0.36, 0.32, 0.14, 0.03, 0.01, 0. ]],

[[ 1. , 0.93, 0.93, 0.88, 0.72, 0.5 , 0.28, 0.03],
[ 0.93, 0.97, 0.99, 0.96, 0.87, 0.7 , 0.42, 0.11],
[ 0.93, 0.99, 0.74, 0.78, 0.78, 0.74, 0.45, 0.2 ],
[ 0.88, 0.96, 0.78, 0. , 0. , 0.78, 0.32, 0.16],
[ 0.72, 0.87, 0.78, 0. , 0. , 0.78, 0.14, 0.06],
[ 0.5 , 0.7 , 0.74, 0.78, 0.78, 0.74, 0.03, 0.01],
[ 0.28, 0.42, 0.45, 0.32, 0.14, 0.03, 0. , 0. ],
[ 0.03, 0.11, 0.2 , 0.16, 0.06, 0.01, 0. , 0. ]],

[[ 1. , 0.91, 0.91, 0.84, 0.64, 0.39, 0.21, 0.03],
[ 0.91, 0.96, 0.98, 0.93, 0.77, 0.53, 0.27, 0.06],
[ 0.91, 0.98, 0.47, 0.5 , 0.5 , 0.47, 0.25, 0.1 ],
[ 0.84, 0.93, 0.5 , 0. , 0. , 0.5 , 0.13, 0.06],
[ 0.64, 0.77, 0.5 , 0. , 0. , 0.5 , 0.03, 0.01],
[ 0.39, 0.53, 0.47, 0.5 , 0.5 , 0.47, 0. , 0. ],
[ 0.21, 0.27, 0.25, 0.13, 0.03, 0. , 0. , 0. ],
[ 0.03, 0.06, 0.1 , 0.06, 0.01, 0. , 0. , 0. ]],

[[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 0. , 0. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 0. , 0. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
[ 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ]]])
r = np.array([[[1.00, 0.95, 0.96, 0.94, 0.86, 0.67, 0.40, 0.03],
[0.95, 0.99, 1.00, 1.00, 1.00, 0.98, 0.67, 0.19],
[0.96, 1.00, 1.00, 1.00, 1.00, 1.00, 0.78, 0.36],
[0.94, 1.00, 1.00, 0.00, 0.00, 1.00, 0.55, 0.32],
[0.86, 1.00, 1.00, 0.00, 0.00, 1.00, 0.27, 0.14],
[0.67, 0.98, 1.00, 1.00, 1.00, 1.00, 0.07, 0.03],
[0.40, 0.67, 0.78, 0.55, 0.27, 0.07, 0.00, 0.01],
[0.03, 0.19, 0.36, 0.32, 0.14, 0.03, 0.01, 0.00]],

[[1.00, 0.93, 0.93, 0.88, 0.72, 0.50, 0.28, 0.03],
[0.93, 0.97, 0.99, 0.96, 0.87, 0.70, 0.42, 0.11],
[0.93, 0.99, 0.74, 0.78, 0.78, 0.74, 0.45, 0.20],
[0.88, 0.96, 0.78, 0.00, 0.00, 0.78, 0.32, 0.16],
[0.72, 0.87, 0.78, 0.00, 0.00, 0.78, 0.14, 0.06],
[0.50, 0.70, 0.74, 0.78, 0.78, 0.74, 0.03, 0.01],
[0.28, 0.42, 0.45, 0.32, 0.14, 0.03, 0.00, 0.00],
[0.03, 0.11, 0.20, 0.16, 0.06, 0.01, 0.00, 0.00]],

[[1.00, 0.91, 0.91, 0.84, 0.64, 0.39, 0.21, 0.03],
[0.91, 0.96, 0.98, 0.93, 0.77, 0.53, 0.27, 0.06],
[0.91, 0.98, 0.47, 0.50, 0.50, 0.47, 0.25, 0.10],
[0.84, 0.93, 0.50, 0.00, 0.00, 0.50, 0.13, 0.06],
[0.64, 0.77, 0.50, 0.00, 0.00, 0.50, 0.03, 0.01],
[0.39, 0.53, 0.47, 0.50, 0.50, 0.47, 0.00, 0.00],
[0.21, 0.27, 0.25, 0.13, 0.03, 0.00, 0.00, 0.00],
[0.03, 0.06, 0.10, 0.06, 0.01, 0.00, 0.00, 0.00]],

[[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 0.00, 0.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 0.00, 0.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00],
[1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00, 1.00]]])
# Result stored transposed and rounded for for more compact display...
assert_array_almost_equal(rgb, r.T, decimal=2)

Expand Down

0 comments on commit 689b3e1

Please sign in to comment.