e2ecdcfe33
Former-commit-id:a02aeb236c[formerly9f19e3f712] [formerlya02aeb236c[formerly9f19e3f712] [formerly06a8b51d6d[formerly 64fa9254b946eae7e61bbc3f513b7c3696c4f54f]]] Former-commit-id:06a8b51d6dFormer-commit-id:8e80217e59[formerly3360eb6c5f] Former-commit-id:377dcd10b9
58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
##
|
|
# This software was developed and / or modified by Raytheon Company,
|
|
# pursuant to Contract DG133W-05-CQ-1067 with the US Government.
|
|
#
|
|
# U.S. EXPORT CONTROLLED TECHNICAL DATA
|
|
# This software product contains export-restricted data whose
|
|
# export/transfer/disclosure is restricted by U.S. law. Dissemination
|
|
# to non-U.S. persons whether in the United States or abroad requires
|
|
# an export license or other authorization.
|
|
#
|
|
# Contractor Name: Raytheon Company
|
|
# Contractor Address: 6825 Pine Street, Suite 340
|
|
# Mail Stop B8
|
|
# Omaha, NE 68106
|
|
# 402.291.0100
|
|
#
|
|
# See the AWIPS II Master Rights File ("Master Rights File.pdf") for
|
|
# further licensing information.
|
|
###
|
|
|
|
|
|
import unittest
|
|
import numpy as np
|
|
import Deformation
|
|
|
|
##
|
|
# Unit test class for Deformation
|
|
class TestDeformation(unittest.TestCase):
|
|
def setUp(self):
|
|
self.U = np.ones((5,5), dtype=np.float32)
|
|
self.U += np.eye(5,5,0)
|
|
self.V = np.ones((5,5), dtype=np.float32)
|
|
|
|
self.V += np.eye(5,5,0).transpose()
|
|
self.Vector = (self.U, self.V)
|
|
self.dx = np.ones((5,5), dtype=np.float32)
|
|
self.dy = np.ones((5,5), dtype=np.float32)
|
|
|
|
def testExecute(self):
|
|
"Test basic results of execute method."
|
|
result = Deformation.execute(self.Vector, self.dx, self.dy)
|
|
correctAnswer = np.array([[1e37, 1e37, 1e37, 1e37, 1e37],
|
|
[1e37, 0, 1, 0, 1e37],
|
|
[1e37, 1, 0, 1, 1e37],
|
|
[1e37, 0, 1, 0, 1e37],
|
|
[1e37, 1e37, 1e37, 1e37, 1e37]], dtype=np.float32)
|
|
self.failUnless(np.allclose(result, correctAnswer), "Wrong answers: " + repr(result))
|
|
|
|
def testMasked(self):
|
|
"Check that invalid data results in cells with ourNaN (1e37)"
|
|
self.Vector[1][1,1] = 1e37
|
|
result = Deformation.execute(self.Vector, self.dx, self.dy)
|
|
correctAnswer = np.array([[1e37, 1e37, 1e37, 1e37, 1e37],
|
|
[1e37, 0, 1e37, 0, 1e37],
|
|
[1e37, 1e37, 0, 1, 1e37],
|
|
[1e37, 0, 1, 0, 1e37],
|
|
[1e37, 1e37, 1e37, 1e37, 1e37]], dtype=np.float32)
|
|
self.failUnless(np.allclose(result, correctAnswer), "Wrong answers: " + repr(result))
|