Merge Grid, SingleGrid, and MultiGrid into one class

examples/Schelling/model.py

@@ -8,6 +8,7 @@ class SchellingAgent(Agent):
     '''
     Schelling segregation agent
     '''
+
     def __init__(self, pos, model, agent_type):
         '''
          Create a new Schelling agent.
@@ -72,7 +73,7 @@ def __init__(self, height=20, width=20, density=0.8, minority_pc=0.2, homophily=
                     agent_type = 0
 
                 agent = SchellingAgent((x, y), self, agent_type)
-                self.grid.position_agent(agent, (x, y))
+                self.grid.position_agent(agent, x, y)
                 self.schedule.add(agent)
 
         self.running = True

mesa/agent.py

@@ -8,11 +8,24 @@
 
 
 class Agent:
-    """ Base class for a model agent. """
+    """ Base class for a model agent. 
+
+    Properties:
+        unique_id: Unique identifer for the agent
+        model: Model that the agent is situated in
+        pos: Position of the agent in the parameter space (if exists)
+    """
+
     def __init__(self, unique_id, model):
-        """ Create a new agent. """
+        """ Create a new agent. 
+
+        Args: 
+            unique_id: Unique identifer for the agent
+            model: Model that the agent is situated in
+        """
         self.unique_id = unique_id
         self.model = model
+        self.pos = None
 
     def step(self):
         """ A single step of the agent. """