ScoDoc-PE/tests/unit/test_pe.py

105 lines
3.3 KiB
Python

"""
Test calcul moyennes pour les poursuites d'études
"""
import numpy as np
import pytest
from tests.unit import setup
from app import db
import app.pe.moys.pe_rcstag as pe_rcstag
@pytest.mark.parametrize(
"notes_S1, notes_S2, coeffs_S1, coeffs_S2, "
"coeffs_rcues_S1, coeffs_rcues_S2, inscr_S1, inscr_S2,"
"moyenne, coeffs_aggreges",
[
pytest.param(
[17.0, 15.0, 12.0],
[16.0, 14.0, 13.0], # notes
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0], # coeffs moy gen
[2.0, 4.0, 6.0],
[8.0, 10.0, 12.0], # coeffs recus
[1.0, 1.0, 1.0],
[1.0, 1.0, 1.0], # inscr
[
(17.0 * 2.0 + 16.0 * 8.0) / (2.0 + 8.0),
(15.0 * 4.0 + 14.0 * 10.0) / (4.0 + 10.0),
(12.0 * 6.0 + 13.0 * 12.0) / (6.0 + 12.0),
],
[1.0 + 4.0, 2.0 + 5.0, 3.0 + 6.0],
id="etudiant_parfait",
),
pytest.param(
[17.0, 15.0, 12.0],
[16.0, 14.0, 13.0], # notes
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0], # coeffs moy gen
[2.0, 4.0, 6.0],
[8.0, 10.0, 12.0], # coeffs recus
[np.nan, 1.0, np.nan],
[1.0, np.nan, np.nan], # inscr
[
(16.0 * 8.0) / (8.0),
(15.0 * 4.0) / (4.0),
np.nan,
],
[4.0, 2.0, np.nan],
id="etudiant_non_inscrit",
),
pytest.param(
[0.0, 15.0, 0.0],
[0.0, 0.0, 0.0], # notes
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0], # coeffs moy gen
[2.0, 4.0, 6.0],
[8.0, 10.0, 12.0], # coeffs recus
[np.nan, 1.0, 1.0],
[1.0, np.nan, 1.0], # inscr
[0.0, 15.0, 0.0],
[4.0, 2.0, 3.0 + 6.0],
id="etudiant_avec_0",
),
],
)
def test_compute_moyennes_par_RCS(
notes_S1,
notes_S2,
coeffs_S1,
coeffs_S2,
coeffs_rcues_S1,
coeffs_rcues_S2,
inscr_S1,
inscr_S2,
moyenne,
coeffs_aggreges,
):
"""Test de pe_rcstag.compute_moyennes_par_RCS"""
notes_cube = np.stack([np.array([notes_S1]), np.array([notes_S2])], axis=-1)
# Vérifie les dimensions
dim1, dim2, dim3 = notes_cube.shape
assert dim1 == 1, "La dim 0 doit être le nombre d'étudiants"
assert dim2 == 3, "La dim 1 doit être le nombre d'UEs/Compétences"
assert dim3 == 2, "La dim 2 doit être le nombre de semestres"
coeffs_cube = np.stack([np.array([coeffs_S1]), np.array([coeffs_S2])], axis=-1)
coeffs_rcue_cube = np.stack(
[np.array([coeffs_rcues_S1]), np.array([coeffs_rcues_S2])], axis=-1
)
inscr_cube = np.stack([np.array([inscr_S1]), np.array([inscr_S2])], axis=-1)
moys, coeffs = pe_rcstag.compute_moyennes_par_RCS(
notes_cube, coeffs_cube, coeffs_rcue_cube, inscr_cube
)
moy_resultat = np.array([moyenne])
assert (
(moys == moy_resultat) | (np.isnan(moys) & np.isnan(moy_resultat))
).all(), "Moyenne erronée"
coeffs_resultat = np.array([coeffs_aggreges])
assert (
(coeffs == coeffs_resultat) | (np.isnan(coeffs) & np.isnan(coeffs_resultat))
).all(), "Coeffs (pour moyenne générale sur toutes les UE) erronés"