Update tests

test/sampler_test.py

@@ -145,7 +145,7 @@ class TestDynesty(unittest.TestCase):
         del self.sampler
 
     def test_default_kwargs(self):
-        expected = dict(bound='multi', sample='rwalk', periodic=None, verbose=True,
+        expected = dict(bound='multi', sample='rwalk', periodic=None, reflective=None, verbose=True,
                         check_point_delta_t=600, nlive=1000, first_update=None,
                         npdim=None, rstate=None, queue_size=None, pool=None,
                         use_pool=None, live_points=None, logl_args=None, logl_kwargs=None,
@@ -155,15 +155,18 @@ class TestDynesty(unittest.TestCase):
                         logl_max=np.inf, add_live=True, print_progress=True, save_bounds=False,
                         walks=20, update_interval=600, print_func='func')
         self.sampler.kwargs['print_func'] = 'func'  # set this manually as this is not testable otherwise
-        self.assertEqual([], self.sampler.kwargs['periodic'])  # Check this separately
-        self.sampler.kwargs['periodic'] = expected['periodic']  # The dict comparison can't handle lists
+        # DictEqual can't handle lists so we check these separately
+        self.assertEqual([], self.sampler.kwargs['periodic'])
+        self.assertEqual([], self.sampler.kwargs['reflective'])
+        self.sampler.kwargs['periodic'] = expected['periodic']
+        self.sampler.kwargs['reflective'] = expected['reflective']
         for key in self.sampler.kwargs.keys():
             print("key={}, expected={}, actual={}"
                   .format(key, expected[key], self.sampler.kwargs[key]))
         self.assertDictEqual(expected, self.sampler.kwargs)
 
     def test_translate_kwargs(self):
-        expected = dict(bound='multi', sample='rwalk', periodic=[], verbose=True,
+        expected = dict(bound='multi', sample='rwalk', periodic=[], reflective=[], verbose=True,
                         check_point_delta_t=600, nlive=1000, first_update=None,
                         npdim=None, rstate=None, queue_size=None, pool=None,
                         use_pool=None, live_points=None, logl_args=None, logl_kwargs=None,
@@ -191,9 +194,10 @@ class TestDynesty(unittest.TestCase):
                                                   outdir='outdir', label='label',
                                                   use_ratio=False, plot=False,
                                                   skip_import_verification=True)
-        self.assertEqual([0, 1, 3, 4], self.sampler.kwargs["periodic"])
-        self.assertEqual([0, 4], self.sampler._periodic)
-        self.assertEqual([1, 3], self.sampler._reflective)
+        self.assertEqual([0, 4], self.sampler.kwargs["periodic"])
+        self.assertEqual(self.sampler._periodic, self.sampler.kwargs["periodic"])
+        self.assertEqual([1, 3], self.sampler.kwargs["reflective"])
+        self.assertEqual(self.sampler._reflective, self.sampler.kwargs["reflective"])
 
 
 class TestEmcee(unittest.TestCase):