black formatting

examples/rabbits/basic/rabbit_agents.py

@@ -5,7 +5,6 @@ import math
 
 
 class RabbitEnv(Environment):
-
     @property
     def num_rabbits(self):
         return self.count_agents(agent_class=Rabbit)
@@ -27,7 +26,7 @@ class Rabbit(NetworkAgent, FSM):
     @default_state
     @state
     def newborn(self):
-        self.info('I am a newborn.')
+        self.info("I am a newborn.")
         self.age = 0
         self.offspring = 0
         return self.youngling
@@ -36,7 +35,7 @@ class Rabbit(NetworkAgent, FSM):
     def youngling(self):
         self.age += 1
         if self.age >= self.sexual_maturity:
-            self.info(f'I am fertile! My age is {self.age}')
+            self.info(f"I am fertile! My age is {self.age}")
             return self.fertile
 
     @state
@@ -60,11 +59,11 @@ class Male(Rabbit):
             return self.dead
 
         # Males try to mate
-        for f in self.model.agents(agent_class=Female,
-                                   state_id=Female.fertile.id,
-                                   limit=self.max_females):
-            self.debug('FOUND A FEMALE: ', repr(f), self.mating_prob)
-            if self.prob(self['mating_prob']):
+        for f in self.model.agents(
+            agent_class=Female, state_id=Female.fertile.id, limit=self.max_females
+        ):
+            self.debug("FOUND A FEMALE: ", repr(f), self.mating_prob)
+            if self.prob(self["mating_prob"]):
                 f.impregnate(self)
                 break  # Take a break
 
@@ -83,14 +82,14 @@ class Female(Rabbit):
             return self.pregnant
 
     def impregnate(self, male):
-        self.info(f'impregnated by {repr(male)}')
+        self.info(f"impregnated by {repr(male)}")
         self.mate = male
         self.pregnancy = 0
-        self.number_of_babies = int(8+4*self.random.random())
+        self.number_of_babies = int(8 + 4 * self.random.random())
 
     @state
     def pregnant(self):
-        self.info('I am pregnant')
+        self.info("I am pregnant")
         self.age += 1
 
         if self.age >= self.life_expectancy:
@@ -100,18 +99,17 @@ class Female(Rabbit):
             self.pregnancy += 1
             return
 
-        self.info('Having {} babies'.format(self.number_of_babies))
+        self.info("Having {} babies".format(self.number_of_babies))
         for i in range(self.number_of_babies):
             state = {}
             agent_class = self.random.choice([Male, Female])
-            child = self.model.add_node(agent_class=agent_class,
-                                        **state)
+            child = self.model.add_node(agent_class=agent_class, **state)
             child.add_edge(self)
             try:
                 child.add_edge(self.mate)
                 self.model.agents[self.mate].offspring += 1
             except ValueError:
-                self.debug('The father has passed away')
+                self.debug("The father has passed away")
 
             self.offspring += 1
         self.mate = None
@@ -119,32 +117,34 @@ class Female(Rabbit):
         return self.fertile
 
     def die(self):
-        if 'pregnancy' in self and self['pregnancy'] > -1:
-            self.info('A mother has died carrying a baby!!')
+        if "pregnancy" in self and self["pregnancy"] > -1:
+            self.info("A mother has died carrying a baby!!")
         return super().die()
 
 
 class RandomAccident(BaseAgent):
-
     def step(self):
         rabbits_alive = self.model.G.number_of_nodes()
 
         if not rabbits_alive:
             return self.die()
 
-        prob_death = self.model.get('prob_death', 1e-100)*math.floor(math.log10(max(1, rabbits_alive)))
-        self.debug('Killing some rabbits with prob={}!'.format(prob_death))
+        prob_death = self.model.get("prob_death", 1e-100) * math.floor(
+            math.log10(max(1, rabbits_alive))
+        )
+        self.debug("Killing some rabbits with prob={}!".format(prob_death))
         for i in self.iter_agents(agent_class=Rabbit):
             if i.state_id == i.dead.id:
                 continue
             if self.prob(prob_death):
-                self.info('I killed a rabbit: {}'.format(i.id))
+                self.info("I killed a rabbit: {}".format(i.id))
                 rabbits_alive -= 1
                 i.die()
-        self.debug('Rabbits alive: {}'.format(rabbits_alive))
+        self.debug("Rabbits alive: {}".format(rabbits_alive))
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     from soil import easy
-    with easy('rabbits.yml') as sim:
+
+    with easy("rabbits.yml") as sim:
         sim.run()

examples/rabbits/improved/rabbit_agents.py

@@ -7,7 +7,6 @@ import math
 
 
 class RabbitEnv(Environment):
-
     @property
     def num_rabbits(self):
         return self.count_agents(agent_class=Rabbit)
@@ -36,7 +35,7 @@ class Rabbit(FSM, NetworkAgent):
     @default_state
     @state
     def newborn(self):
-        self.info('I am a newborn.')
+        self.info("I am a newborn.")
         self.birth = self.now
         self.offspring = 0
         return self.youngling, Delta(self.sexual_maturity - self.age)
@@ -44,7 +43,7 @@ class Rabbit(FSM, NetworkAgent):
     @state
     def youngling(self):
         if self.age >= self.sexual_maturity:
-            self.info(f'I am fertile! My age is {self.age}')
+            self.info(f"I am fertile! My age is {self.age}")
             return self.fertile
 
     @state
@@ -66,11 +65,11 @@ class Male(Rabbit):
             return self.dead
 
         # Males try to mate
-        for f in self.model.agents(agent_class=Female,
-                                   state_id=Female.fertile.id,
-                                   limit=self.max_females):
-            self.debug('FOUND A FEMALE: ', repr(f), self.mating_prob)
-            if self.prob(self['mating_prob']):
+        for f in self.model.agents(
+            agent_class=Female, state_id=Female.fertile.id, limit=self.max_females
+        ):
+            self.debug("FOUND A FEMALE: ", repr(f), self.mating_prob)
+            if self.prob(self["mating_prob"]):
                 f.impregnate(self)
                 break  # Do not try to impregnate other females
 
@@ -94,32 +93,31 @@ class Female(Rabbit):
         return self.now - self.conception
 
     def impregnate(self, male):
-        self.info(f'impregnated by {repr(male)}')
+        self.info(f"impregnated by {repr(male)}")
         self.mate = male
         self.conception = self.now
-        self.number_of_babies = int(8+4*self.random.random())
+        self.number_of_babies = int(8 + 4 * self.random.random())
 
     @state
     def pregnant(self):
-        self.debug('I am pregnant')
+        self.debug("I am pregnant")
 
         if self.age > self.life_expectancy:
             self.info("Dying before giving birth")
             return self.die()
 
         if self.pregnancy >= self.gestation:
-            self.info('Having {} babies'.format(self.number_of_babies))
+            self.info("Having {} babies".format(self.number_of_babies))
             for i in range(self.number_of_babies):
                 state = {}
                 agent_class = self.random.choice([Male, Female])
-                child = self.model.add_node(agent_class=agent_class,
-                                            **state)
+                child = self.model.add_node(agent_class=agent_class, **state)
                 child.add_edge(self)
                 if self.mate:
                     child.add_edge(self.mate)
                     self.mate.offspring += 1
                 else:
-                    self.debug('The father has passed away')
+                    self.debug("The father has passed away")
 
                 self.offspring += 1
             self.mate = None
@@ -127,31 +125,33 @@ class Female(Rabbit):
 
     def die(self):
         if self.pregnancy is not None:
-            self.info('A mother has died carrying a baby!!')
+            self.info("A mother has died carrying a baby!!")
         return super().die()
 
 
 class RandomAccident(BaseAgent):
-
     def step(self):
         rabbits_alive = self.model.G.number_of_nodes()
 
         if not rabbits_alive:
             return self.die()
 
-        prob_death = self.model.get('prob_death', 1e-100)*math.floor(math.log10(max(1, rabbits_alive)))
-        self.debug('Killing some rabbits with prob={}!'.format(prob_death))
+        prob_death = self.model.get("prob_death", 1e-100) * math.floor(
+            math.log10(max(1, rabbits_alive))
+        )
+        self.debug("Killing some rabbits with prob={}!".format(prob_death))
         for i in self.iter_agents(agent_class=Rabbit):
             if i.state_id == i.dead.id:
                 continue
             if self.prob(prob_death):
-                self.info('I killed a rabbit: {}'.format(i.id))
+                self.info("I killed a rabbit: {}".format(i.id))
                 rabbits_alive -= 1
                 i.die()
-        self.debug('Rabbits alive: {}'.format(rabbits_alive))
+        self.debug("Rabbits alive: {}".format(rabbits_alive))
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     from soil import easy
-    with easy('rabbits.yml') as sim:
+
+    with easy("rabbits.yml") as sim:
         sim.run()