0.13.8

tests/test_examples.py

@@ -7,6 +7,8 @@ from soil import utils, simulation
 ROOT = os.path.abspath(os.path.dirname(__file__))
 EXAMPLES = join(ROOT, '..', 'examples')
 
+FORCE_TESTS = os.environ.get('FORCE_TESTS', '')
+
 
 class TestExamples(TestCase):
     pass
@@ -19,7 +21,10 @@ def make_example_test(path, config):
         s = simulation.from_config(config)
         iterations = s.max_time * s.num_trials
         if iterations > 1000:
-            self.skipTest('This example would probably take too long')
+            s.max_time = 100
+            s.num_trials = 1
+        if config.get('skip_test', False) and not FORCE_TESTS:
+            self.skipTest('Example ignored.')
         envs = s.run_simulation(dry_run=True)
         assert envs
         for env in envs:

tests/test_main.py

@@ -320,3 +320,19 @@ class TestMain(TestCase):
         h = history.History()
         h.save_record(agent_id=0, t_step=0, key="test", value="hello")
         assert h[0, 0, "test"] == "hello"
+
+    def test_subgraph(self):
+        '''An agent should be able to subgraph the global topology'''
+        G = nx.Graph()
+        G.add_node(3)
+        G.add_edge(1, 2)
+        distro = agents.calculate_distribution(agent_type=agents.NetworkAgent)
+        env = Environment(name='Test', topology=G, network_agents=distro)
+        lst = list(env.network_agents)
+
+        a2 = env.get_agent(2)
+        a3 = env.get_agent(3)
+        assert len(a2.subgraph(limit_neighbors=True)) == 2
+        assert len(a3.subgraph(limit_neighbors=True)) == 1
+        assert len(a3.subgraph(limit_neighbors=True, center=False)) == 0
+        assert len(a3.subgraph(agent_type=agents.NetworkAgent)) == 3