def create_test_simulation_output(self): np.random.seed(1) cluster = clusters.SimpleCluster() simulation = BaseLdfSimulation(cluster, self.data, '/sim', R=100, N=100) self.redirect_stdout_stderr_to_devnull() simulation.run() self.restore_stdout_stderr()
def create_test_simulation_output(self): np.random.seed(1) self.sim = 'E_100TeV/zenith_0' cluster = clusters.SimpleCluster(size=50) simulation = GroundParticlesSimulation(cluster, self.data, os.path.join('/showers', self.sim, 'leptons'), # os.path.join('/simulations', self.sim), force=True, os.path.join('/test_output', self.sim), R=10, N=20) self.redirect_stdout_stderr_to_devnull() simulation.run() self.restore_stdout_stderr()
def test_get_coordinates_after_init(self): cluster = clusters.SimpleCluster() coordinates = cluster.get_xyalpha_coordinates() self.assertEqual(coordinates, (0., 0., 0.))
def test_init_calls_super_init(self): with patch.object(clusters.BaseCluster, '__init__', mocksignature=True) as mock_base_init: clusters.SimpleCluster() self.assertTrue(mock_base_init.called)
def test_cluster_stations(self): cluster = clusters.SimpleCluster() stations = cluster.stations self.assertEqual(len(stations), 4)
def setUp(self): self.cluster = clusters.SimpleCluster(size=100)