test_writeRaster.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Sep 28 16:43:25 2021

@author: geopeter
"""
import SQLPandasTools as s2pTool
import write_raster as cut
import unittest
import pickle

class TestWriteRaster(unittest.TestCase):
    def testcalcAverageYear(self):
        with open("./pickle/averageDataForRaster.pickle", "rb") as pickleFile:
            average_data = pickle.load(pickleFile)
            
            ramp = [[255, 255, 255, 1], [255, 244, 191, 1], [255, 233, 128, 1], [255, 221, 64, 1], [255, 210, 0, 1], [243, 105, 0, 1], [230, 0, 0, 1], [153, 0, 0, 1], [77, 0, 0, 1], [0, 0, 0, 1]]
            
            cut.write_raster(average_data, ramp)
            
            
if __name__ == '__main__':
    unittest.main()