def test_error_unmatched_weights(self): """Test error when weights shape doesn't match length of blend dimension (in this case 3 weights for 2 blend slices)""" weights = generate_matching_weights_array([0.7, 0.1, 0.2], 1) percentiles = np.array([0, 50, 100]) perc_data = np.array([[1.0, 2.0], [5.0, 5.0], [10.0, 9.0]]) with self.assertRaisesRegex(ValueError, "Weights shape does not match data"): PercentileBlendingAggregator.aggregate(perc_data, 1, percentiles, weights)
def test_blend_percentile_aggregate(self): """Test blend_percentile_aggregate function works""" weights = np.array([0.8, 0.2]) percentiles = np.array([0, 20, 40, 60, 80, 100]) result = PercentileBlendingAggregator.aggregate( np.reshape(PERCENTILE_DATA, (6, 2, 2, 2)), 1, percentiles, weights, 0) expected_result_array = np.reshape(BLENDED_PERCENTILE_DATA2, (6, 2, 2)) self.assertArrayAlmostEqual(result, expected_result_array)
def test_3D_simple_case(self): """ Test that for a simple case with only one point and an extra internal dimension behaves as expected""" weights = np.array([0.5, 0.5]) percentiles = np.array([0, 50, 100]) perc_data = np.array([[[1.0], [2.0]], [[5.0], [6.0]], [[10.0], [9.0]]]) result = PercentileBlendingAggregator.aggregate( perc_data, 1, percentiles, weights, 0) expected_result = np.array([[1.0], [5.555555], [10.0]]) self.assertArrayAlmostEqual(result, expected_result)
def test_2D_simple_case(self): """ Test that for a simple case with only one point in the resulting array the function behaves as expected""" weights = np.array([0.8, 0.2]) percentiles = np.array([0, 50, 100]) perc_data = np.array([[1.0, 2.0], [5.0, 5.0], [10.0, 9.0]]) result = PercentileBlendingAggregator.aggregate( perc_data, 1, percentiles, weights, 0) expected_result = np.array([1.0, 5.0, 10.0]) self.assertArrayAlmostEqual(result, expected_result)
def test_blend_percentile_aggregate(self): """Test blend_percentile_aggregate function works""" weights = generate_matching_weights_array([0.6, 0.3, 0.1], 4) percentiles = np.array([0, 20, 40, 60, 80, 100]).astype(np.float32) result = PercentileBlendingAggregator.aggregate( PERCENTILE_DATA, 1, percentiles, weights) self.assertArrayAlmostEqual( result, BLENDED_PERCENTILE_DATA, )
def test_blend_percentile_aggregate_reorder2(self): """Test blend_percentile_aggregate works with out of order dims 2""" weights = np.array([0.8, 0.2]) percentiles = np.array([0, 20, 40, 60, 80, 100]) perc_data = np.reshape(PERCENTILE_DATA, (6, 2, 2, 2)) perc_data = np.moveaxis(perc_data, [0, 1], [1, 2]) result = PercentileBlendingAggregator.aggregate( perc_data, 2, percentiles, weights, 1) expected_result_array = np.reshape(BLENDED_PERCENTILE_DATA2, (6, 2, 2)) expected_result_array = np.moveaxis(expected_result_array, 0, 1) self.assertArrayAlmostEqual(result, expected_result_array)
def test_blend_percentile_aggregate(self): """Test blend_percentile_aggregate function works""" weights = np.array([0.6, 0.3, 0.1]) weights = generate_matching_weights_array(weights, (4, 6, 3)) weights = np.moveaxis(weights, (0, 1, 2), (2, 1, 0)) percentiles = np.array([0, 20, 40, 60, 80, 100]).astype(np.float32) result = PercentileBlendingAggregator.aggregate( np.reshape(PERCENTILE_DATA, (6, 3, 2, 2)), 1, percentiles, weights, 0 ) self.assertArrayAlmostEqual( result, BLENDED_PERCENTILE_DATA, )
def test_4D_simple_case(self): """ Test that for a simple case with only one point and 4D input data it behaves as expected""" weights = np.array([0.5, 0.5]) percentiles = np.array([0, 50, 100]) perc_data = np.array([1.0, 3.0, 2.0, 4.0, 5.0, 6.0]) input_shape = (3, 2, 1, 1) perc_data = perc_data.reshape(input_shape) result = PercentileBlendingAggregator.aggregate( perc_data, 1, percentiles, weights, 0) expected_result = np.array([[[1.0]], [[3.5]], [[6.0]]]) expected_result_shape = (3, 1, 1) self.assertArrayAlmostEqual(result, expected_result) self.assertEqual(result.shape, expected_result_shape)
def test_blend_percentile_aggregate_reorder2(self): """Test blend_percentile_aggregate works with out of order dims 2""" weights = np.array([0.6, 0.3, 0.1]) weights = generate_matching_weights_array(weights, (4, 6, 3)) weights = np.moveaxis(weights, (0, 1, 2), (2, 1, 0)) percentiles = np.array([0, 20, 40, 60, 80, 100]) perc_data = np.reshape(PERCENTILE_DATA, (6, 3, 2, 2)) perc_data = np.moveaxis(perc_data, [0, 1], [1, 2]) result = PercentileBlendingAggregator.aggregate( perc_data, 2, percentiles, weights, 1) expected_result_array = BLENDED_PERCENTILE_DATA expected_result_array = np.moveaxis(expected_result_array, 0, 1) self.assertArrayAlmostEqual(result, expected_result_array)