diff --git a/tests/test_algorithm.py b/tests/test_algorithm.py index 0b1601e035074d3e0ade722290d5325977153380..dd876fbf437f76832b0b22b2a52a22722237c50e 100644 --- a/tests/test_algorithm.py +++ b/tests/test_algorithm.py @@ -240,7 +240,7 @@ def test_gmm(): np.testing.assert_allclose( algorithm.machine.means, array( - [[3.246973, 2.413918], [4.382181, 4.336439], [0.44701, 3.653619]] + [[5.269755, 1.545634], [3.275648, 2.999225], [0.44701, 3.653619]] ), atol=1e-5, rtol=1e-5, @@ -249,8 +249,8 @@ def test_gmm(): algorithm.machine.variances, array( [ - [9.384035e-01, 6.412975e-01], - [2.287441e-02, 1.781911e-02], + [2.220446e-16, 2.220446e-16], + [6.359690e-01, 1.046621e00], [2.220446e-16, 2.220446e-16], ] ), @@ -259,7 +259,7 @@ def test_gmm(): ) np.testing.assert_allclose( algorithm.machine.weights, - array([0.701606, 0.198394, 0.1]), + array([0.1, 0.8, 0.1]), atol=1e-5, rtol=1e-5, ) @@ -267,16 +267,16 @@ def test_gmm(): fused, array( [ - -23.2301, - -15.178111, - -15.79934, - -14.814705, - -19.907042, - -14.284925, - -19.53618, - -10.727596, - -10.803899, - -15.976103, + -25.888355, + -21.89579, + -18.766963, + -20.338103, + -25.582336, + -18.796124, + -21.475942, + -13.596667, + -13.428485, + -19.276524, ] ), atol=1e-5,