def test_get_coordinates_after_init(self): cluster = clusters.SingleTwoDetectorStation() coordinates = cluster.get_xyalpha_coordinates() self.assertEqual(coordinates, (0., 0., 0.))
def test_single_station(self): cluster = clusters.SingleTwoDetectorStation() stations = cluster.stations self.assertEqual(len(stations), 1) detectors = stations[0].detectors self.assertEqual(len(detectors), 2)
def test_init_calls_super_init(self): with patch.object(clusters.BaseCluster, '__init__', mocksignature=True) as mock_base_init: clusters.SingleTwoDetectorStation() self.assertTrue(mock_base_init.called)