Example #1
0
 def testGraph(self):
     return
     import networkx as nx
     graph = nx.geographical_threshold_graph(100,1.2)
     self.g = GraphSpace(graph)
     for idx, node in self.g.cells():
         self.g[idx] = sim.BugCell()
     self.bugs = sim.init_grid(self.g, numbugs=20)
Example #2
0
class SimTest(unittest.TestCase):
    def setUp(self):
        self.sim = sim.Heatbugs()

    def testCellAdd(self):
        bug = self.sim.add_bug()
        self.assertTrue(bug is not None)

    def testCellOverAdd(self):
        bug = True
        index, cell = self.sim.grid.cells().next()
        for x in range(self.sim.properties.max_bugs_per_cell + 1):
            bug = self.sim.add_bug(index)
        self.assertTrue(bool(bug) == False)

    def testSimStep(self):
        return
        for bug in self.bugs:
            self.sim.step_bug(bug)

    def testGraph(self):
        return
        import networkx as nx
        graph = nx.geographical_threshold_graph(100,1.2)
        self.g = GraphSpace(graph)
        for idx, node in self.g.cells():
            self.g[idx] = sim.BugCell()
        self.bugs = sim.init_grid(self.g, numbugs=20)

    def testGraphStep(self):
        return
        self.testSimStep()

    def testRunSim(self):
        return
        sim.run_simulation(steps=1)