Added a test for the weights

Author: mesonepigreco

tests/test_rho_update/rho.dat

@@ -0,0 +1,100 @@
+4.539220355821488306e-01
+1.411526449794866522e+00
+2.104710325189953046e+00
+1.307973163516430670e+00
+9.698244944445526539e-01
+6.079171591146247389e-01
+8.638305258146310450e-01
+1.364249855976386483e+00
+1.150362638627647982e+00
+9.282376473677631523e-01
+7.481192048339360134e-01
+3.963286781380914725e-01
+8.697536941689273382e-01
+8.246715668237457608e-01
+3.994984794357198110e-01
+5.046320360558806950e-01
+9.329591782510011644e-01
+1.034134496933409197e+00
+1.261538977433160902e+00
+8.983074250578575448e-01
+1.016582863091882283e+00
+1.178658097285228301e+00
+5.400118861260385650e-01
+5.815053907267632161e-01
+1.672259771808280115e+00
+1.193477382384837915e+00
+1.139073730384374539e+00
+8.978012988414395545e-01
+8.891090775619920894e-01
+2.459552753806575343e+00
+9.824445587590747708e-01
+1.157391371610775321e+00
+1.359971082419375144e+00
+1.186643630233218172e+00
+7.227648865314558746e-01
+1.095252449457546851e+00
+5.454369201695307146e-01
+1.104938252792029285e+00
+7.482719193837685800e-01
+1.498463555239989420e+00
+5.572994625789204903e-01
+7.705556253595574612e-01
+6.986817913535011693e-01
+5.447730186037899047e-01
+1.141109810306783379e+00
+9.919214313253675863e-01
+8.552194270819423183e-01
+8.114897140621727401e-01
+7.490298145732271484e-01
+5.733037139914664015e-01
+7.814040834364547328e-01
+1.184479312351917502e+00
+1.431239307147331496e+00
+5.745975497809883148e-01
+6.852131914998749185e-01
+7.331097689064051126e-01
+4.424517778140502067e-01
+1.648708815350376167e+00
+6.498338247002786172e-01
+8.659896174386515488e-01
+6.691119551407023192e-01
+5.392713915162969940e-01
+1.251206844589672906e+00
+4.362932779004319306e-01
+5.922883410805960258e-01
+8.794036099552269281e-01
+9.019277098138261950e-01
+1.160878890503636240e+00
+1.105650006125480811e+00
+1.165694660516127801e+00
+9.486042209926186519e-01
+5.014128121316550057e-01
+1.082562669335646488e+00
+6.828508512622004645e-01
+6.216703952241426911e-01
+6.891275313437438266e-01
+9.977369856535546200e-01
+7.427079255525675938e-01
+7.694943795491897554e-01
+1.113645044986756760e+00
+1.560928717016897282e+00
+7.277430228963687187e-01
+5.139526369854987564e-01
+1.004312304546424173e+00
+3.358750079123379861e-01
+4.447819855182386095e-01
+1.216612868580458917e+00
+4.800835463817738069e-01
+6.242423793254461284e-01
+9.930756363113935059e-01
+4.709592911549231875e-01
+1.031033707302008828e+00
+1.234728079118472044e+00
+1.648877984075169723e+00
+1.387104853294882867e+00
+6.636840015273967053e-01
+7.958957361111163031e-01
+8.739496782321736212e-01
+8.012650880673536369e-01
+6.980785469953588729e-01

tests/test_rho_update/test_rho_update.py

@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+
+"""
+"""
+from __future__ import print_function
+from __future__ import division
+
+import sys, os
+
+import numpy as np
+import cellconstructor as CC
+import cellconstructor.Phonons
+
+import sscha, sscha.Ensemble, sscha.Utilities
+import sscha.SchaMinimizer
+
+def test_rho_update():
+    total_path = os.path.dirname(os.path.abspath(__file__))
+    os.chdir(total_path)
+
+    # Load the starting dynamical matrix
+    dyn = CC.Phonons.Phonons("../../Examples/ensemble_data_test/dyn")
+
+    # Load the ensemble
+    N_RAND = 100
+    T0 = 0
+    POP = 2
+    EQ_ENERGY = -144.40680397
+    MEANINGFUL = 0.001
+
+    ens = sscha.Ensemble.Ensemble(dyn, T0)
+    ens.load("../../Examples/ensemble_data_test", POP, N_RAND)
+
+    final_dyn = CC.Phonons.Phonons("../../Examples/ensemble_data_test/dyn1_population2", full_name = True)
+    
+    ens.update_weights(final_dyn, T0)
+
+    rho_true = np.loadtxt("rho.dat")
+    assert np.max(np.abs(rho_true - ens.rho)) < 1e-6, "Error, the weights are wrong."
+    
+
+if __name__ == "__main__":
+    test_rho_update()
+