Fixed matplotlib.use() issue.

tests/test_kmeans.py

@@ -10,13 +10,9 @@ Tests for gms_preprocessing.algorithms.L2B_P.KMeansRSImage
 
 import unittest
 import os
-import matplotlib
 import numpy as np
-
 from sklearn.cluster import k_means_
 
-matplotlib.use('Template')  # disables matplotlib figure popups
-
 from geoarray import GeoArray  # noqa E402 module level import not at top of file
 
 from gms_preprocessing import __file__  # noqa E402 module level import not at top of file
@@ -39,6 +35,8 @@ class Test_KMeansRSImage(unittest.TestCase):
         cls.geoArr.to_mem()
         cls.kmeans = KMeansRSImage(cls.geoArr, n_clusters=10)
 
+        os.environ['MPLBACKEND'] = 'Template'  # disables matplotlib figure popups # NOTE: import geoarray sets 'Agg'
+
     def test_compute_clusters(self):
         self.kmeans.compute_clusters()
         self.assertIsInstance(self.kmeans.clusters, k_means_.KMeans)