remove unused imports and fix tests

Author: aloctavodia

pymc3/step_methods/smc.py

@@ -13,7 +13,6 @@
 @author: Hannes Vasyura-Bathke
 """
 import numpy as np
-from math import log
 import pymc3 as pm
 from tqdm import tqdm
 

pymc3/tests/test_step.py

@@ -124,56 +124,56 @@ class TestStepMethods(object):  # yield test doesn't work subclassing object
               -2.24238542e+00,  -1.01648100e+00,  -1.01648100e+00,  -7.60912865e-01,
                1.44384812e+00,   2.07355127e+00,   1.91390340e+00,   1.66559696e+00]),
         smc.SMC: np.array(
-      [-6.93241492e-01, -5.28171752e-01, -6.92278431e-01, -2.43673631e-01,
-       -3.63864239e-01, -6.02622986e-01, -1.76442235e-01, -4.74737493e-01,
-       -1.70834637e-01, -4.74737493e-01, -1.13549112e-01, -4.74737493e-01,
-       -1.31522036e-01, -4.68991109e-01,  7.63062953e-04, -2.95225417e-01,
-        9.97982970e-02, -3.26875232e-01,  6.68871285e-01, -1.16513174e-01,
-        3.87738482e-01,  6.40532105e-03, -7.80174876e-03, -7.86970315e-01,
-        4.33184436e-02, -1.08233426e+00,  4.03565846e-01, -1.23289460e+00,
-        2.69998078e-01, -1.46323431e+00,  2.04993681e-01, -1.50992195e+00,
-       -3.33970215e-01, -1.21973608e+00, -7.88408084e-01, -1.41565214e+00,
-       -5.84146424e-02, -1.25518887e+00, -4.80486087e-01, -1.17872185e+00,
-       -8.18802072e-01, -8.62776390e-01, -9.40225900e-01, -2.35403303e-01,
-       -9.69990002e-01, -3.48602791e-01, -9.69990002e-01, -2.79247411e-01,
-       -7.89632438e-02, -1.05937822e+00, -4.09611000e-01, -1.31747706e+00,
-       -3.66674846e-01, -1.31747706e+00, -7.94202588e-01, -1.17432908e+00,
-       -4.80532287e-01, -1.17432908e+00, -1.77206900e-01, -7.96723348e-01,
-       -7.28509985e-01, -9.57924049e-01, -7.28509985e-01, -1.45709101e+00,
-       -1.21378465e+00, -1.32457141e+00, -1.31496879e+00, -4.86639174e-01,
-       -1.17887328e+00, -6.89999248e-01, -1.35028409e+00, -1.03303266e-01,
-       -1.12312406e+00, -1.44621146e-01, -4.57723003e-01, -3.53424989e-01,
-       -3.92354632e-01, -3.34353526e-01, -9.96278538e-01, -7.31361439e-02,
-       -1.19947059e+00, -6.45642112e-02, -1.18808814e+00,  2.27888805e-01,
-       -1.09906934e+00, -1.96855532e-01, -1.29447893e+00, -1.02132533e+00,
-       -1.20127000e+00, -7.94592757e-01, -5.75119699e-01, -4.16947040e-01,
-        2.63758223e-02, -1.81726404e-01, -5.31717327e-01, -4.68493741e-01,
-       -6.55849127e-01, -7.21904775e-01, -1.52970012e+00, -3.02813116e-01,
-       -1.50894460e+00, -3.02813116e-01, -1.28748459e+00, -6.02637106e-03,
-       -1.43364184e+00,  8.58605578e-01, -1.72678130e+00,  7.85718430e-01,
-       -1.84590731e+00,  1.03110215e+00, -1.58240871e+00,  8.39376856e-01,
-       -1.58240871e+00,  1.14362740e+00, -1.15731486e+00,  2.18966211e-02,
-       -1.07616310e+00, -3.31309057e-01, -1.23388873e+00, -8.92389246e-01,
-       -1.62202943e+00, -8.83638309e-01, -1.63034094e+00, -9.92076558e-01,
-       -1.50965041e+00, -1.31305091e+00, -1.50965041e+00, -1.31305091e+00,
-       -1.65726150e+00, -1.28007002e+00, -1.65726150e+00, -1.29773750e+00,
-       -1.94687988e+00, -1.51365263e+00, -2.03542695e+00, -1.54872720e+00,
-       -2.03542695e+00, -1.76912251e+00, -1.73858008e+00, -1.97496533e+00,
-       -1.78821545e+00, -2.16123061e+00, -2.13443513e+00, -2.16123061e+00,
-       -2.03231814e+00, -2.16123061e+00, -2.45380923e+00, -2.16123061e+00,
-       -2.61662278e+00, -1.65836396e+00, -2.51266355e+00, -1.50303503e+00,
-       -2.45715919e+00, -1.50303503e+00, -2.45715919e+00, -1.58349936e+00,
-       -2.45715919e+00, -1.58349936e+00, -2.43398707e+00, -1.57696134e+00,
-       -2.43398707e+00, -1.46192791e+00, -2.40033545e+00, -1.28419657e+00,
-       -1.76386224e+00, -1.41739935e+00, -1.91555315e+00, -1.31916129e+00,
-       -1.54761255e+00, -1.31966101e+00, -1.54761255e+00, -1.31067298e+00,
-       -1.22332511e+00, -1.27278477e+00, -1.22332511e+00, -1.27187793e+00,
-       -1.35273944e+00, -1.33384497e+00, -1.16478821e+00, -1.37384162e+00,
-       -1.04170331e+00, -8.36595290e-01, -1.28980912e+00, -1.07938479e+00,
-       -1.61713227e+00, -1.01319959e+00, -1.68580351e+00, -5.81263343e-01,
-       -2.11296992e+00, -1.36766786e+00, -2.08456777e+00, -1.28086329e+00,
-       -2.08399471e+00, -1.28086329e+00, -2.28961132e+00, -1.29666202e+00,
-       -2.36879548e+00, -1.13213387e+00, -2.40904612e+00, -1.14895523e+00]),
+      [-6.93241492e-01, -7.31306682e-01, -6.92278431e-01, -5.38971903e-01,
+       -3.63864239e-01, -7.81639393e-01, -1.76442235e-01, -6.95182484e-01,
+       -1.70834637e-01, -6.91297645e-01, -1.13549112e-01, -5.73823456e-01,
+       -1.31522036e-01, -4.05141577e-01,  7.63062953e-04, -3.11034353e-01,
+        9.97982970e-02, -1.56511470e-01,  6.68871285e-01, -1.15387717e+00,
+        3.87738482e-01, -1.19314682e+00, -7.80174876e-03, -1.43622635e+00,
+        4.33184436e-02, -1.93210174e+00,  4.03565846e-01, -1.93210174e+00,
+        2.69998078e-01, -1.63371961e+00,  2.04993681e-01, -1.49152889e+00,
+       -3.33970215e-01, -1.07618355e+00, -7.88408084e-01, -1.07618355e+00,
+       -5.84146424e-02, -7.60238090e-01, -4.80486087e-01, -1.32865002e-01,
+       -8.18802072e-01, -2.09393447e-01, -9.40225900e-01, -1.62505785e-01,
+       -9.69990002e-01, -3.71928783e-01, -9.69990002e-01, -1.79001101e-02,
+       -7.89632438e-02, -1.79001101e-02, -4.09611000e-01,  7.88749920e-02,
+       -3.66674846e-01,  2.05150150e-01, -7.94202588e-01,  4.98861576e-01,
+       -4.80532287e-01,  3.37660876e-01, -1.77206900e-01, -1.61506088e-01,
+       -7.28509985e-01,  1.73439211e-01, -7.28509985e-01,  3.47487801e-02,
+       -9.56879974e-01, -1.59855934e-02, -7.24175290e-01, -2.72376580e-01,
+       -1.01682490e+00, -2.48958662e-01, -1.21307819e+00,  7.17910968e-02,
+       -1.67818076e+00,  7.89068592e-02, -1.98491030e+00,  3.21678968e-01,
+       -1.43814668e+00, -3.09112690e-02, -1.43814668e+00, -7.15322991e-01,
+       -1.84024408e+00, -4.88590415e-01, -1.51673992e+00, -1.10944697e-01,
+       -1.45569455e+00,  1.24275938e-01, -1.34887321e+00, -1.62491398e-01,
+       -1.14603705e+00, -4.15902432e-01, -1.11446001e+00,  3.18922609e-03,
+       -1.34720657e+00, -3.39934926e-01, -1.70915288e+00, -4.31481807e-02,
+       -1.63971745e+00,  8.21483768e-01, -1.63752467e+00,  7.48596620e-01,
+       -1.63752467e+00,  9.93980340e-01, -1.90938468e+00,  8.02255046e-01,
+       -2.02030962e+00,  1.10650559e+00, -2.31344909e+00, -1.52251885e-02,
+       -2.43257510e+00, -3.68430866e-01, -2.16907649e+00, -9.29511056e-01,
+       -2.16907649e+00, -9.18969326e-01, -1.74398265e+00, -1.04959839e+00,
+       -1.66283089e+00, -1.43625696e+00, -1.82055651e+00, -1.43625696e+00,
+       -2.13665682e+00, -1.40327607e+00, -2.02511907e+00, -1.42094355e+00,
+       -2.03288017e+00, -1.63685868e+00, -2.21063115e+00, -1.67193325e+00,
+       -2.33677066e+00, -1.89232856e+00, -2.12598116e+00, -2.09817138e+00,
+       -2.12598116e+00, -2.28443667e+00, -2.33324231e+00, -2.28443667e+00,
+       -2.41326691e+00, -2.28443667e+00, -2.34146132e+00, -2.28443667e+00,
+       -2.18383170e+00, -1.95866523e+00, -2.24356939e+00, -1.85803869e+00,
+       -1.86676826e+00, -1.79709743e+00, -1.88295378e+00, -2.02148945e+00,
+       -1.86017722e+00, -2.02148945e+00, -1.86017722e+00, -2.07361656e+00,
+       -1.86017722e+00, -1.79664076e+00, -2.11641422e+00, -1.57798197e+00,
+       -2.06769020e+00, -1.26676256e+00, -2.02636926e+00, -1.05742456e+00,
+       -1.38989604e+00, -9.87916736e-01, -1.54158695e+00, -9.77082655e-01,
+       -1.17364636e+00, -8.75178531e-01, -1.23600577e+00, -8.72739504e-01,
+       -1.49246121e+00, -8.72739504e-01, -1.69458186e+00, -1.12213269e+00,
+       -1.37029442e+00, -1.09619745e+00, -1.28062186e+00, -1.09619745e+00,
+       -1.39608733e+00, -1.09619745e+00, -1.20813610e+00, -1.48145119e+00,
+       -9.64693905e-01, -1.18526483e+00, -6.85433431e-01, -1.03724907e+00,
+       -6.85433431e-01, -1.03724907e+00, -6.32619521e-01, -9.90567503e-01,
+       -6.31553924e-01, -1.10462261e+00, -8.25572702e-01, -1.17635498e+00,
+       -8.25572702e-01, -1.17181362e+00, -9.00418925e-01, -1.07920544e+00,
+       -7.09372067e-01, -7.85138291e-01, -5.89973809e-01, -6.47909120e-01]),
     }
 
     def setup_class(self):
@@ -214,6 +214,7 @@ def check_trace(self, step_method):
                                progressbar=False,
                                step=step_method(),
                                step_kwargs={'homepath': self.temp_dir})
+
             elif step_method.__name__ == 'NUTS':
                 step = step_method(scaling=model.test_point)
                 trace = sample(0, tune=n_steps,