Theory citation function

flavio/test_citations.py

@@ -8,15 +8,35 @@ def test_citations(self):
         self.assertIn("Straub:2018kue", citations.set)
         flavio.sm_prediction("DeltaGamma_s")
         self.assertIn("Beneke:2003az", citations.set)
+        citations.register("fakename:2020abc")
+        self.assertIn("fakename:2020abc", citations.set)
         citations.reset()
         self.assertNotIn("Beneke:2003az", citations.set)
+        self.assertNotIn("fakename:2020abc", citations.set)
+        self.assertIn("Straub:2018kue", citations.set)
 
-    def test_register(self):
+    def test_clear_citations(self):
         citations = flavio.citations
         citations.register("fakename:2020abc")
-        self.assertIn("fakename:2020abc", citations.set)
+        citations.clear()
+        self.assertNotIn("fakename:2020abc", citations.set)
+        self.assertNotIn("Straub:2018kue", citations.set)
 
     def test_theory_citations(self):
         DGs_citations = flavio.Observable["DeltaGamma_s"].theory_citations()
         self.assertNotIn("Straub:2018kue", DGs_citations)
         self.assertIn("Beneke:2003az", DGs_citations)
+
+    def test_bad_inspirekey(self):
+        with self.assertRaises(KeyError):
+            flavio.citations.register("bad:inspirekey")
+
+    def test_multithread(self):
+        obs = ("DeltaGamma_s", "m_W")
+        flavio.citations.reset()
+        flavio.sm_covariance(obs, threads=1)
+        cites_singlethread = flavio.citations.set
+        flavio.citations.reset()
+        flavio.sm_covariance(obs, threads=4)
+        cites_multithread = flavio.citations.set
+        self.assertSetEqual(cites_singlethread, cites_multithread)