ITS: add legend to test_raw_exposure.py plots
bug: 235174961
Change-Id: I66dbff22451988c981918f913e5081b002b17578
diff --git a/apps/CameraITS/tests/scene1_2/test_raw_exposure.py b/apps/CameraITS/tests/scene1_2/test_raw_exposure.py
index 5bd44e8..bcf426d 100644
--- a/apps/CameraITS/tests/scene1_2/test_raw_exposure.py
+++ b/apps/CameraITS/tests/scene1_2/test_raw_exposure.py
@@ -85,15 +85,16 @@
gb = [m[2] for m in means[1:]]
b = [m[3] for m in means[1:]]
pylab.figure('%s_%s' % (NAME, sens))
- pylab.plot(exps, r, 'r.-')
- pylab.plot(exps, b, 'b.-')
- pylab.plot(exps, gr, 'g.-')
- pylab.plot(exps, gb, 'k.-')
+ pylab.plot(exps, r, 'r.-', label='R')
+ pylab.plot(exps, gr, 'g.-', label='Gr')
+ pylab.plot(exps, gb, 'k.-', label='Gb')
+ pylab.plot(exps, b, 'b.-', label='B')
pylab.xscale('log')
pylab.yscale('log')
pylab.title('%s ISO=%d' % (NAME, sens))
pylab.xlabel('Exposure time (ms)')
pylab.ylabel('Center patch pixel mean')
+ pylab.legend(loc='lower right', numpoints=1, fancybox=True)
matplotlib.pyplot.savefig(
'%s_s=%d.png' % (os.path.join(log_path, NAME), sens))
pylab.clf()
