From 39f557bb202c84bfed97bc841cd27eea74b57a33 Mon Sep 17 00:00:00 2001 From: Dimitra Salmanidou Date: Mon, 10 Jun 2024 20:13:35 +0100 Subject: [PATCH] style fixes --- src/hivpy/hiv_status.py | 3 +-- src/tests/test_hiv_status.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/hivpy/hiv_status.py b/src/hivpy/hiv_status.py index 36634d5..e26e7e8 100644 --- a/src/hivpy/hiv_status.py +++ b/src/hivpy/hiv_status.py @@ -262,7 +262,7 @@ def stp_HIV_transmission(self, person): def ltp_transmission(self, population: Population): """ - Sets population set for which HIV transmission occurs from a long term partner. + Sets population set for which HIV transmission occurs from a long term partner. Considers monogamous and non-monogamous relationships. """ num_neg_women_with_infected_parter = len(population.get_sub_pop([(col.SEX, op.eq, SexType.Female), @@ -443,7 +443,6 @@ def update_HIV_status(self, population: Population): # TODO: Add ltp HIV transmission # Determine HIV status after transmission new_HIV_status = population.apply_function(self.stp_HIV_transmission, 1, HIV_neg_active_pop) - # Apply HIV status to sub-population population.set_present_variable(col.HIV_STATUS, new_HIV_status, diff --git a/src/tests/test_hiv_status.py b/src/tests/test_hiv_status.py index 3fb0fef..d72e689 100644 --- a/src/tests/test_hiv_status.py +++ b/src/tests/test_hiv_status.py @@ -408,7 +408,7 @@ def test_no_infection_in_nonmonogamous_ltp(): def test_ltp_deterministic_variables(): """ - Check that for a specific group in the population the estimation is as expected + Check that for a specific group in the population the estimation is as expected """ N = 1000 pop = Population(size=N, start_date=date(1990, 1, 1))