def setUp(self): TestCase.setUp(self) self.t1 = Transition(29, 9, 4) distributions = {} enf = np.ones((5, 5, 5)) distributions[PhotonKey(self.t1, True, PhotonKey.P)] = enf self.r = PhotonEmissionMapResult(distributions)
class TestPhotonEmissionMapResult(TestCase): def setUp(self): TestCase.setUp(self) self.t1 = Transition(29, 9, 4) distributions = {} enf = np.ones((5, 5, 5)) distributions[PhotonKey(self.t1, True, PhotonKey.P)] = enf self.r = PhotonEmissionMapResult(distributions) def testexists(self): self.assertTrue(self.r.exists(self.t1)) self.assertTrue(self.r.exists('Cu La1')) self.assertFalse(self.r.exists('Cu Ka1')) def testget(self): m = self.r.get(self.t1, True, False) self.assertEqual(3, m.ndim)