def testProblemCalcBasisGradient(self): coords = np.array([[0.,0.],[ 0.0625,0. ],[ 0.,0.0625]]) bg = Problem._calcBasisGradient(coords) self.assertTrue((bg == np.array([[512.,-256.,-256.],[-256.,256.,0.], [-256.,0.,256.]])).all())