Fix and add a test

test/gw_prior_test.py

@@ -3,6 +3,7 @@ from collections import OrderedDict
 import unittest
 import os
 import sys
+import pickle
 
 import numpy as np
 from astropy import cosmology
@@ -110,6 +111,16 @@ class TestBBHPriorDict(unittest.TestCase):
             minimum=20, maximum=40, name='chirp_mass')
         self.assertFalse(self.bbh_prior_dict.test_has_redundant_keys())
 
+    def test_pickle_prior(self):
+        priors = dict(chirp_mass=bilby.core.prior.Uniform(10, 20),
+                      mass_ratio=bilby.core.prior.Uniform(0.125, 1))
+        priors = bilby.gw.prior.BBHPriorDict(priors)
+        with open("test.pickle", "wb") as file:
+            pickle.dump(priors, file)
+        with open("test.pickle", "rb") as file:
+            priors_loaded = pickle.load(file)
+        self.assertEqual(priors, priors_loaded)
+
 
 class TestPackagedPriors(unittest.TestCase):
     """ Test that the prepackaged priors load """

test/prior_test.py

@@ -658,8 +658,8 @@ class TestPriorDict(unittest.TestCase):
         priors_set = bilby.core.prior.PriorSet(self.priors)
         self.assertEqual(priors_dict, priors_set)
 
-    def test_prior_set_is_ordered_dict(self):
-        self.assertIsInstance(self.prior_set_from_dict, OrderedDict)
+    def test_prior_set_is_dict(self):
+        self.assertIsInstance(self.prior_set_from_dict, dict)
 
     def test_prior_set_has_correct_length(self):
         self.assertEqual(3, len(self.prior_set_from_dict))