def test_weights_get_functions(self): """Tests the preprocessing for weights function""" import numpy basisfn, dim = interpolator._get_basis_functions(['L1'], None) self.assertEqual(1, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions(['L1'], None) self.assertEqual(1, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions(['L1'], None) self.assertEqual(1, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions( ['L1', 'L3'], None) self.assertEqual(2, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) self.assertEqual('L3', basisfn[1][0].__name__) self.assertEqual([1], basisfn[1][1]) basisfn, dim = interpolator._get_basis_functions( ['L1', 'T22'], None) self.assertEqual(2, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) self.assertEqual('T22', basisfn[1][0].__name__) self.assertEqual([1, 2], basisfn[1][1]) basisfn, dim = interpolator._get_basis_functions( ['T44', 'L2'], None) self.assertEqual(2, len(basisfn)) self.assertEqual('T44', basisfn[0][0].__name__) self.assertEqual([0, 1], basisfn[0][1]) self.assertEqual('L2', basisfn[1][0].__name__) self.assertEqual([2], basisfn[1][1])
def test_weights_get_functions(self): """Tests the preprocessing for weights function""" import numpy basisfn, dim = interpolator._get_basis_functions(['L1'], None) self.assertEqual(1, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions(['L1'], None) self.assertEqual(1, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions(['L1'], None) self.assertEqual(1, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions(['L1', 'L3'], None) self.assertEqual(2, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) self.assertEqual('L3', basisfn[1][0].__name__) self.assertEqual([1], basisfn[1][1]) basisfn, dim = interpolator._get_basis_functions(['L1', 'T22'], None) self.assertEqual(2, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) self.assertEqual('T22', basisfn[1][0].__name__) self.assertEqual([1, 2], basisfn[1][1]) basisfn, dim = interpolator._get_basis_functions(['T44', 'L2'], None) self.assertEqual(2, len(basisfn)) self.assertEqual('T44', basisfn[0][0].__name__) self.assertEqual([0, 1], basisfn[0][1]) self.assertEqual('L2', basisfn[1][0].__name__) self.assertEqual([2], basisfn[1][1])
def test_weights_get_functions_deriv(self): """Tests the preprocessing for weights function""" import numpy basisfn, dim = interpolator._get_basis_functions(['L1'], deriv=[0]) self.assertEqual(1, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions(['L1'], deriv=[1]) self.assertEqual(1, len(basisfn)) self.assertEqual('L1d1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions( ['L1', 'L2', 'L3', 'L4'], deriv=[0, 0, 0, 0]) self.assertEqual(4, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) self.assertEqual('L2', basisfn[1][0].__name__) self.assertEqual([1], basisfn[1][1]) self.assertEqual('L3', basisfn[2][0].__name__) self.assertEqual([2], basisfn[2][1]) self.assertEqual('L4', basisfn[3][0].__name__) self.assertEqual([3], basisfn[3][1]) basisfn, dim = interpolator._get_basis_functions( ['L1', 'L2', 'L3', 'L4'], deriv=[1, 1, 1, 1]) self.assertEqual(4, len(basisfn)) self.assertEqual('L1d1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) self.assertEqual('L2d1', basisfn[1][0].__name__) self.assertEqual([1], basisfn[1][1]) self.assertEqual('L3d1', basisfn[2][0].__name__) self.assertEqual([2], basisfn[2][1]) self.assertEqual('L4d1', basisfn[3][0].__name__) self.assertEqual([3], basisfn[3][1]) basisfn, dim = interpolator._get_basis_functions(['L1'], deriv=[2]) self.assertEqual(1, len(basisfn)) self.assertEqual('L1d1d1', basisfn[0][0].__name__) basisfn, dim = interpolator._get_basis_functions(['H3'], deriv=[0]) self.assertEqual(1, len(basisfn)) self.assertEqual('H3', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions(['H3'], deriv=[1]) self.assertEqual(1, len(basisfn)) self.assertEqual('H3d1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions(['H3'], deriv=[2]) self.assertEqual(1, len(basisfn)) self.assertEqual('H3d1d1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions( ['T11', 'T22', 'T33', 'T44'], deriv=[0, 0, 0, 0, 0, 0, 0, 0]) self.assertEqual(4, len(basisfn)) self.assertEqual('T11', basisfn[0][0].__name__) self.assertEqual([0, 1], basisfn[0][1]) self.assertEqual('T22', basisfn[1][0].__name__) self.assertEqual([2, 3], basisfn[1][1]) self.assertEqual('T33', basisfn[2][0].__name__) self.assertEqual([4, 5], basisfn[2][1]) self.assertEqual('T44', basisfn[3][0].__name__) self.assertEqual([6, 7], basisfn[3][1]) basisfn, dim = interpolator._get_basis_functions(['T44', 'T44'], deriv=[1, 0, 0, 1]) self.assertEqual(2, len(basisfn)) self.assertEqual('T44d1', basisfn[0][0].__name__) self.assertEqual([0, 1], basisfn[0][1]) self.assertEqual('T44d2', basisfn[1][0].__name__) self.assertEqual([2, 3], basisfn[1][1]) basisfn, dim = interpolator._get_basis_functions(['T44', 'L2'], deriv=[1, 0, 1]) self.assertEqual(2, len(basisfn)) self.assertEqual('T44d1', basisfn[0][0].__name__) self.assertEqual([0, 1], basisfn[0][1]) self.assertEqual('L2d1', basisfn[1][0].__name__) self.assertEqual([2], basisfn[1][1])
def test_weights_get_functions_deriv(self): """Tests the preprocessing for weights function""" import numpy basisfn, dim = interpolator._get_basis_functions(['L1'], deriv=[0]) self.assertEqual(1, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions(['L1'], deriv=[1]) self.assertEqual(1, len(basisfn)) self.assertEqual('L1d1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions( ['L1', 'L2', 'L3', 'L4'], deriv=[0, 0, 0, 0]) self.assertEqual(4, len(basisfn)) self.assertEqual('L1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) self.assertEqual('L2', basisfn[1][0].__name__) self.assertEqual([1], basisfn[1][1]) self.assertEqual('L3', basisfn[2][0].__name__) self.assertEqual([2], basisfn[2][1]) self.assertEqual('L4', basisfn[3][0].__name__) self.assertEqual([3], basisfn[3][1]) basisfn, dim = interpolator._get_basis_functions( ['L1', 'L2', 'L3', 'L4'], deriv=[1, 1, 1, 1]) self.assertEqual(4, len(basisfn)) self.assertEqual('L1d1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) self.assertEqual('L2d1', basisfn[1][0].__name__) self.assertEqual([1], basisfn[1][1]) self.assertEqual('L3d1', basisfn[2][0].__name__) self.assertEqual([2], basisfn[2][1]) self.assertEqual('L4d1', basisfn[3][0].__name__) self.assertEqual([3], basisfn[3][1]) basisfn, dim = interpolator._get_basis_functions( ['L1'], deriv=[2]) self.assertEqual(1, len(basisfn)) self.assertEqual('L1d1d1', basisfn[0][0].__name__) basisfn, dim = interpolator._get_basis_functions( ['H3'], deriv=[0]) self.assertEqual(1, len(basisfn)) self.assertEqual('H3', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions( ['H3'], deriv=[1]) self.assertEqual(1, len(basisfn)) self.assertEqual('H3d1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions( ['H3'], deriv=[2]) self.assertEqual(1, len(basisfn)) self.assertEqual('H3d1d1', basisfn[0][0].__name__) self.assertEqual([0], basisfn[0][1]) basisfn, dim = interpolator._get_basis_functions( ['T11', 'T22', 'T33', 'T44'], deriv=[0, 0, 0, 0, 0, 0, 0, 0]) self.assertEqual(4, len(basisfn)) self.assertEqual('T11', basisfn[0][0].__name__) self.assertEqual([0, 1], basisfn[0][1]) self.assertEqual('T22', basisfn[1][0].__name__) self.assertEqual([2, 3], basisfn[1][1]) self.assertEqual('T33', basisfn[2][0].__name__) self.assertEqual([4, 5], basisfn[2][1]) self.assertEqual('T44', basisfn[3][0].__name__) self.assertEqual([6, 7], basisfn[3][1]) basisfn, dim = interpolator._get_basis_functions( ['T44', 'T44'], deriv=[1, 0, 0, 1]) self.assertEqual(2, len(basisfn)) self.assertEqual('T44d1', basisfn[0][0].__name__) self.assertEqual([0, 1], basisfn[0][1]) self.assertEqual('T44d2', basisfn[1][0].__name__) self.assertEqual([2, 3], basisfn[1][1]) basisfn, dim = interpolator._get_basis_functions( ['T44', 'L2'], deriv=[1, 0, 1]) self.assertEqual(2, len(basisfn)) self.assertEqual('T44d1', basisfn[0][0].__name__) self.assertEqual([0, 1], basisfn[0][1]) self.assertEqual('L2d1', basisfn[1][0].__name__) self.assertEqual([2], basisfn[1][1])