add tests for cockroft-gault and mdrd

Author: nerdoc

tests/test_cg.py

@@ -0,0 +1,133 @@
+import pytest
+
+from medimetry.constants import Gender
+from medimetry.renal import cockcroft_gault
+
+
+def test_cockcroft_gault_male_normal():
+    """Test Cockcroft-Gault formula for normal male patient."""
+    # 70-year-old male, 70kg, creatinine 1.0 mg/dl
+    result = cockcroft_gault(age=70, weight=70, creatinine=1.0, gender=Gender.MALE)
+    expected = round(((140 - 70) * 70) / (72 * 1.0))  # 68 mL/min
+    assert result == expected
+    assert result == 68
+
+
+def test_cockcroft_gault_female_normal():
+    """Test Cockcroft-Gault formula for normal female patient."""
+    # 70-year-old female, 60kg, creatinine 1.0 mg/dl
+    result = cockcroft_gault(age=70, weight=60, creatinine=1.0, gender=Gender.FEMALE)
+    expected = round(((140 - 70) * 60) / (72 * 1.0) * 0.85)  # 49 mL/min
+    assert result == expected
+    assert result == 50
+
+
+def test_cockcroft_gault_young_patient():
+    """Test Cockcroft-Gault formula for young patient."""
+    # 30-year-old male, 80kg, creatinine 0.8 mg/dl
+    result = cockcroft_gault(age=30, weight=80, creatinine=0.8, gender=Gender.MALE)
+    expected = round(((140 - 30) * 80) / (72 * 0.8))  # 153 mL/min
+    assert result == expected
+    assert result == 153
+
+
+def test_cockcroft_gault_elderly_patient():
+    """Test Cockcroft-Gault formula for elderly patient."""
+    # 85-year-old female, 55kg, creatinine 1.5 mg/dl
+    result = cockcroft_gault(age=85, weight=55, creatinine=1.5, gender=Gender.FEMALE)
+    expected = round(((140 - 85) * 55) / (72 * 1.5) * 0.85)  # 25 mL/min
+    assert result == expected
+    assert result == 24
+
+
+def test_cockcroft_gault_high_creatinine():
+    """Test Cockcroft-Gault formula with high creatinine."""
+    # 60-year-old male, 75kg, creatinine 3.0 mg/dl
+    result = cockcroft_gault(age=60, weight=75, creatinine=3.0, gender=Gender.MALE)
+    expected = round(((140 - 60) * 75) / (72 * 3.0))  # 28 mL/min
+    assert result == expected
+    assert result == 28
+
+
+def test_cockcroft_gault_zero_weight():
+    """Test Cockcroft-Gault formula with zero weight."""
+    with pytest.raises(AssertionError, match="Weight must be positive"):
+        cockcroft_gault(age=50, weight=0, creatinine=1.0, gender=Gender.MALE)
+
+
+def test_cockcroft_gault_negative_weight():
+    """Test Cockcroft-Gault formula with negative weight."""
+    with pytest.raises(AssertionError, match="Weight must be positive"):
+        cockcroft_gault(age=50, weight=-10, creatinine=1.0, gender=Gender.MALE)
+
+
+def test_cockcroft_gault_excessive_weight():
+    """Test Cockcroft-Gault formula with excessive weight."""
+    with pytest.raises(AssertionError, match="Weight must be less than 400 kg"):
+        cockcroft_gault(age=50, weight=450, creatinine=1.0, gender=Gender.MALE)
+
+
+def test_cockcroft_gault_zero_age():
+    """Test Cockcroft-Gault formula with zero age."""
+    with pytest.raises(AssertionError, match="Age must be positive"):
+        cockcroft_gault(age=0, weight=70, creatinine=1.0, gender=Gender.MALE)
+
+
+def test_cockcroft_gault_negative_age():
+    """Test Cockcroft-Gault formula with negative age."""
+    with pytest.raises(AssertionError, match="Age must be positive"):
+        cockcroft_gault(age=-5, weight=70, creatinine=1.0, gender=Gender.MALE)
+
+
+def test_cockcroft_gault_negative_creatinine():
+    """Test Cockcroft-Gault formula with negative creatinine."""
+    with pytest.raises(AssertionError, match="Creatinine must not be negative"):
+        cockcroft_gault(age=50, weight=70, creatinine=-0.5, gender=Gender.MALE)
+
+
+def test_cockcroft_gault_zero_creatinine():
+    """Test Cockcroft-Gault formula with zero creatinine."""
+    # Zero creatinine should be allowed (though clinically unrealistic)
+    with pytest.raises(ValueError, match="Creatinine must be non-zero"):
+        cockcroft_gault(age=50, weight=70, creatinine=0.0, gender=Gender.MALE)
+    # This would result in division by zero, so the function should handle this
+
+
+def test_cockcroft_gault_invalid_gender():
+    """Test Cockcroft-Gault formula with invalid gender."""
+    with pytest.raises(AssertionError, match="gender must be a Gender instance"):
+        cockcroft_gault(age=50, weight=70, creatinine=1.0, gender="invalid")
+    with pytest.raises(AssertionError, match="gender must be a Gender instance"):
+        cockcroft_gault(age=50, weight=70, creatinine=1.0, gender=2)
+
+
+def test_cockcroft_gault_realistic_scenarios():
+    """Test Cockcroft-Gault formula with realistic clinical scenarios."""
+    # Healthy young adult male
+    result = cockcroft_gault(age=25, weight=75, creatinine=0.9, gender=Gender.MALE)
+    assert result > 100  # Should have good kidney function
+
+    # Elderly female with mild kidney impairment
+    result = cockcroft_gault(age=80, weight=60, creatinine=1.3, gender=Gender.FEMALE)
+    assert 30 < result < 60  # Moderate kidney impairment
+
+    # Middle-aged male with normal function
+    result = cockcroft_gault(age=45, weight=80, creatinine=1.1, gender=Gender.MALE)
+    assert 60 < result < 120  # Normal to mildly reduced
+
+
+def test_cockcroft_gault_gender_difference():
+    """Test that female patients have lower clearance than male patients."""
+    male_result = cockcroft_gault(age=60, weight=70, creatinine=1.0, gender=Gender.MALE)
+    female_result = cockcroft_gault(age=60, weight=70, creatinine=1.0, gender=Gender.FEMALE)
+
+    # Female result should be 85% of male result
+    expected_female = round(male_result * 0.85)
+    assert female_result < male_result
+    assert abs(female_result - expected_female) <= 1  # Allow for rounding differences
+
+
+def test_cockcroft_gault_return_type():
+    """Test that Cockcroft-Gault returns an integer."""
+    result = cockcroft_gault(age=50, weight=70, creatinine=1.0, gender=Gender.MALE)
+    assert isinstance(result, int)

tests/test_mdrd.py

@@ -0,0 +1,84 @@
+import pytest
+
+from medimetry import EthnicalRace
+from medimetry import Gender
+from medimetry.renal import mdrd
+
+
+def test_mdrd_male_normal_creatinine():
+    """Test MDRD formula for male patient with normal creatinine levels."""
+    # 50-year-old male, creatinine 1.0 mg/dl, other race
+    result = mdrd(creatinine=1.0, age=50, gender=Gender.MALE, race=EthnicalRace.OTHER)
+    expected = 175 * (1.0**-1.154) * (50**-0.203)  # ~79.7 mL/min/1.73m²
+    assert abs(result - expected) < 0.1
+    assert 75 < result < 85  # Should be in normal range
+
+
+def test_mdrd_female_correction_factor():
+    """Test MDRD formula applies 0.742 correction factor for female patients."""
+    # Test with same parameters but different gender to verify correction factor
+    male_result = mdrd(creatinine=1.0, age=50, gender=Gender.MALE, race=EthnicalRace.OTHER)
+    female_result = mdrd(creatinine=1.0, age=50, gender=Gender.FEMALE, race=EthnicalRace.OTHER)
+
+    # Female result should be male result * 0.742
+    expected_female = male_result * 0.742
+    assert abs(female_result - expected_female) < 0.001  # Allow for floating point precision
+    assert female_result < male_result  # Female result should be lower
+
+
+def test_mdrd_african_american_correction():
+    """Test MDRD formula with African American race correction factor."""
+    # 50-year-old African American male, creatinine 1.2 mg/dl
+    result = mdrd(creatinine=1.2, age=50, gender=Gender.MALE, race=EthnicalRace.AFRICAN_AMERICAN)
+
+    # Calculate expected value: 175 * (1.2)^-1.154 * (50)^-0.203 * 1.212
+    base_calculation = 175 * (1.2**-1.154) * (50**-0.203)
+    expected = base_calculation * 1.212
+
+    assert abs(result - expected) < 0.01  # Allow small floating point differences
+    assert result > base_calculation  # African American factor should increase result
+
+
+def test_mdrd_african_american_female():
+    """Test MDRD formula for African American female with both correction factors."""
+    # 50-year-old African American female, creatinine 1.2 mg/dl
+    result = mdrd(creatinine=1.2, age=50, gender=Gender.FEMALE, race=EthnicalRace.AFRICAN_AMERICAN)
+    # Base: 175 * (1.2)^-1.154 * (50)^-0.203 = 175 * 0.8676 * 0.4472 = 67.88
+    # Female factor: 67.88 * 0.742 = 50.37
+    # African American factor: 50.37 * 1.212 = 61.05
+    expected = 57.6
+    assert abs(result - expected) < 0.1
+
+
+def test_mdrd_zero_creatinine():
+    """Test MDRD formula with zero creatinine."""
+    with pytest.raises(AssertionError, match="Creatinine must be positive"):
+        mdrd(creatinine=0.0, age=50, gender=Gender.MALE)
+
+
+def test_mdrd_negative_creatinine():
+    """Test MDRD formula with negative creatinine."""
+    with pytest.raises(AssertionError, match="Creatinine must be positive"):
+        mdrd(creatinine=-0.5, age=50, gender=Gender.MALE)
+
+
+def test_mdrd_zero_age():
+    """Test MDRD formula with zero age."""
+    with pytest.raises(AssertionError, match="Age must be positive"):
+        mdrd(creatinine=1.0, age=0, gender=Gender.MALE)
+
+
+def test_mdrd_negative_age():
+    """Test MDRD formula with negative age."""
+    with pytest.raises(AssertionError, match="Age must be positive"):
+        mdrd(creatinine=1.0, age=-5, gender=Gender.MALE)
+
+
+def test_mdrd_elderly_over_100():
+    """Test MDRD formula for elderly patients over 100 years."""
+    # 105-year-old male, creatinine 1.2 mg/dl
+    result = mdrd(creatinine=1.2, age=105, gender=Gender.MALE)
+    # Manual calculation: 175 * (1.2)^-1.154 * (105)^-0.203
+    # = 175 * 0.8551 * 0.3147 ≈ 47.1
+    assert 55.1 < result < 55.2
+    assert isinstance(result, float)