github.com/network-quality/goresponsiveness@v0.0.0-20240129151524-343954285090/qualityattenuation/qualityattenuation_test.go (about)

     1  /*
     2   * This file is part of Go Responsiveness.
     3   *
     4   * Go Responsiveness is free software: you can redistribute it and/or modify it under
     5   * the terms of the GNU General Public License as published by the Free Software Foundation,
     6   * either version 2 of the License, or (at your option) any later version.
     7   * Go Responsiveness is distributed in the hope that it will be useful, but WITHOUT ANY
     8   * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
     9   * PARTICULAR PURPOSE. See the GNU General Public License for more details.
    10   *
    11   * You should have received a copy of the GNU General Public License along
    12   * with Go Responsiveness. If not, see <https://www.gnu.org/licenses/>.
    13   */
    14  
    15  package qualityattenuation
    16  
    17  import (
    18  	"testing"
    19  
    20  	"github.com/stretchr/testify/assert"
    21  )
    22  
    23  func TestBasicSimpleQualityAttenuation(t *testing.T) {
    24  	qa := NewSimpleQualityAttenuation()
    25  	qa.AddSample(1.0)
    26  	qa.AddSample(2.0)
    27  	qa.AddSample(3.0)
    28  	assert.Equal(t, qa.numberOfSamples, int64(3))
    29  	assert.Equal(t, qa.numberOfLosses, int64(0))
    30  	assert.InEpsilon(t, 1.0, qa.minimumLatency, 0.000001)
    31  	assert.InEpsilon(t, 3.0, qa.maximumLatency, 0.000001)
    32  	assert.InEpsilon(t, 5.7, qa.offsetSum, 0.000001)
    33  	assert.InEpsilon(t, 12.83, qa.offsetSumOfSquares, 0.000001)
    34  	assert.InEpsilon(t, 1.0, qa.empiricalDistribution.Quantile(0.1), 0.000001)
    35  	assert.InEpsilon(t, 2.0, qa.empiricalDistribution.Quantile(0.5), 0.000001)
    36  	assert.InEpsilon(t, 3.0, qa.empiricalDistribution.Quantile(0.9), 0.000001)
    37  	// Test the get functions
    38  	assert.Equal(t, qa.GetNumberOfSamples(), int64(3))
    39  	assert.Equal(t, qa.GetNumberOfLosses(), int64(0))
    40  	assert.InEpsilon(t, 1.0, qa.GetMinimum(), 0.000001)
    41  	assert.InEpsilon(t, 3.0, qa.GetMaximum(), 0.000001)
    42  	assert.InEpsilon(t, 2.0, qa.GetAverage(), 0.000001)
    43  	assert.InEpsilon(t, 1.000000, qa.GetVariance(), 0.000001)
    44  	assert.InEpsilon(t, 1.000000, qa.GetStandardDeviation(), 0.000001)
    45  	assert.InEpsilon(t, 2.0, qa.GetMedian(), 0.000001)
    46  	assert.InEpsilon(t, 1.0, qa.GetLossPercentage()+1.000000000, 0.000001)
    47  	assert.InEpsilon(t, 30, qa.GetRPM(), 0.000001)
    48  	assert.InEpsilon(t, 1.0, qa.GetPercentile(10.0), 0.000001)
    49  	assert.InEpsilon(t, 2.0, qa.GetPercentile(50.0), 0.000001)
    50  	assert.InEpsilon(t, 3.0, qa.GetPercentile(90.0), 0.000001)
    51  	assert.InEpsilon(t, 2.0, qa.GetPDV(90), 0.000001)
    52  }
    53  
    54  func TestManySamples(t *testing.T) {
    55  	qa := NewSimpleQualityAttenuation()
    56  	for i := 1; i < 160000; i++ {
    57  		qa.AddSample(float64(i) / 10000.0) // Linear ramp from 0.0001 to 16.0
    58  	}
    59  	assert.Equal(t, qa.numberOfSamples, int64(160000-1))
    60  	assert.Equal(t, qa.numberOfLosses, int64(10000-1)) // Samples from 15.0001 to 16.0 are lost
    61  	assert.InEpsilon(t, 0.0001, qa.minimumLatency, 0.000001)
    62  	assert.InEpsilon(t, 15.0000, qa.maximumLatency, 0.000001)
    63  	assert.InEpsilon(t, 1110007.5, qa.offsetSum, 0.000001)
    64  	assert.InEpsilon(t, 11026611.00024998, qa.offsetSumOfSquares, 0.000001)
    65  	assert.InEpsilon(t, 1.50005, qa.empiricalDistribution.Quantile(0.1), 0.000001)
    66  	assert.InEpsilon(t, 7.500049, qa.empiricalDistribution.Quantile(0.5), 0.000001)
    67  	assert.InEpsilon(t, 13.50005, qa.empiricalDistribution.Quantile(0.9), 0.000001)
    68  	// Test the get functions
    69  	assert.Equal(t, qa.GetNumberOfSamples(), int64(160000-1))
    70  	assert.Equal(t, qa.GetNumberOfLosses(), int64(10000-1))
    71  	assert.InEpsilon(t, 0.0001, qa.GetMinimum(), 0.000001)
    72  	assert.InEpsilon(t, 15.0000, qa.GetMaximum(), 0.000001)
    73  	assert.InEpsilon(t, 7.500049, qa.GetAverage(), 0.000001)
    74  	assert.InEpsilon(t, 18.750120, qa.GetVariance(), 0.000001)
    75  	assert.InEpsilon(t, 4.330141, qa.GetStandardDeviation(), 0.000001)
    76  	assert.InEpsilon(t, 7.500049, qa.GetMedian(), 0.000001)
    77  	assert.InEpsilon(t, 6.249414, qa.GetLossPercentage(), 0.000001)
    78  	assert.InEpsilon(t, 7.999947, qa.GetRPM(), 0.000001)
    79  }