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)
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)