added plot routine to population

population-density/population_file_extraction.py

@@ -8,8 +8,8 @@ Author: Martin Schultz, FZ Juelich (04 May 2016)
 import numpy as np
 import os
 from toar_location_services.settings import DATA_DIR, DEBUG, USE_DUMMY_POPULATION_DATA
+import matplotlib.pyplot as plt
 
-# import matplotlib.pyplot as plt
 
 def read_proxydata(filename, dummy=DEBUG and USE_DUMMY_POPULATION_DATA):
     """Read the ascii file and return the data array together with
@@ -54,9 +54,11 @@ def read_proxydata(filename, dummy=DEBUG and USE_DUMMY_POPULATION_DATA):
 
     logdata = data.copy()
     logdata[logdata <= 1.e-4] = 1.e-4
-    # plt.contourf(lonvec, latvec, np.log10(logdata))
-    # plt.savefig('global_population_density.png')
-    # plt.close()
+
+    if DEBUG:
+        plt.contourf(lonvec, latvec, np.log10(logdata))
+        plt.savefig('../plots/global_population_density.png')
+        plt.close()
 
     # set metadata
     boundingbox = [lon0, lat0, lonvec.max(), latvec.max()]

population-density/tests.py

 from django.test import TestCase
 
 # Create your tests here.
+
+from .population_file_extraction import read_proxydata
+
+
+FILENAME = "pop.txt"
+
+read_proxydata(FILENAME, dummy=False)
\ No newline at end of file