def test_get_uint8_image(self): ''' Shall scale image values from float (or any) to 0 - 255 [uint8] ''' imageUint8 = get_uint8_image(self.n1['sigma0_HV'], self.imgMin, self.imgMax, 0, 0) plt.imsave('sea_ice_drift_tests_%s.png' % inspect.currentframe().f_code.co_name, imageUint8, vmin=0, vmax=255) self.assertEqual(imageUint8.dtype, np.uint8) self.assertEqual(imageUint8.min(), 1) self.assertEqual(imageUint8.max(), 255)
def setUp(self): ''' Load test data ''' testDir = os.getenv('ICE_DRIFT_TEST_DATA_DIR') if testDir is None: sys.exit('ICE_DRIFT_TEST_DATA_DIR is not defined') testFiles = glob.glob(os.path.join(testDir, 'S1?_*tif')) if len(testFiles) < 2: sys.exit('Not enough test files in %s' % testDir) # sort by date dates = [os.path.basename(f).split('_')[4] for f in testFiles] self.testFiles = [str(f) for f in np.array(testFiles)[np.argsort(dates)]] self.n1 = Nansat(self.testFiles[0]) self.n2 = Nansat(self.testFiles[1]) self.imgMin = 0.001 self.imgMax = 0.013 self.nFeatures = 5000 self.img1 = get_uint8_image(self.n1['sigma0_HV'], self.imgMin, self.imgMax, 0, 0) self.img2 = get_uint8_image(self.n2['sigma0_HV'], self.imgMin, self.imgMax, 0, 0)