def test_calculate_charge_balance_correctly(self):
# sample = nwisaz.01.92600003
expected_balance = 0.27
expected_cations = 10.45
expected_anions = 10.39
dep = Concentration()
dep._set('Bicarbonate', 188, None)
dep._set('Calcium', 66, None)
dep._set('Chloride', 57, None)
dep._set('Magnesium', 27, None)
dep._set('Nitrate', 0.8, None)
dep._set('Potassium', 7.4, None)
dep._set('Sodium', 109, None)
dep._set('Sulfate', 273, None)
balance, cations, anions = self.patient.calculate_charge_balance(dep)
self.assertEqual(balance, expected_balance)
self.assertEqual(cations, expected_cations)
self.assertEqual(anions, expected_anions)
# sample = nwisaz.01.92600006
expected_balance = -0.02
expected_cations = 4.21
expected_anions = 4.21
dep = Concentration()
dep._set('Bicarbonate', 139, None)
dep._set('Calcium', 46, None)
dep._set('Chloride', 12, None)
dep._set('Magnesium', 10, None)
dep._set('Nitrate', 0.5, None)
dep._set('Sodium plus potassium', 25, None)
dep._set('Sulfate', 76, None)
balance, cations, anions = self.patient.calculate_charge_balance(dep)
self.assertEqual(balance, expected_balance)
self.assertEqual(cations, expected_cations)
self.assertEqual(anions, expected_anions)
