← Back to branch summary

~madteam/mg5amcnlo/series2.0

« back to all changes in this revision

Viewing changes to tests/unit_tests/core/test_helas_objects.py

mwrge with 2.7.3

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/unit_tests/core/test_helas_objects.py
13
13
#
14
14
################################################################################
15
15
from __future__ import absolute_import
 
16
from __future__ import print_function
16
17
from madgraph.iolibs import helas_call_writers
17
18
from six.moves import range
18
19
from six.moves import zip
3747
3748
            sym = proc.identical_particle_factor()
3748
3749
            if self.debugging:
3749
3750
                if sol == sym:
3750
 
                    print("OK  :  %s: expected %s returned %s" % (key, sol, sym))
 
3751
                    print(("OK  :  %s: expected %s returned %s" % (key, sol, sym)))
3751
3752
                else:
3752
 
                    print("FAIL  :  %s: expected %s returned %s" % (key, sol, sym))
 
3753
                    print(("FAIL  :  %s: expected %s returned %s" % (key, sol, sym)))
3753
3754
                    passed = False
3754
3755
            else:
3755
3756
                self.assertEqual(sol, sym, " %s: expected %s returned %s" % (key, sol, sym))
3768
3769
            sym = proc.identical_particle_factor()
3769
3770
            if self.debugging:
3770
3771
                if sol == sym:
3771
 
                    print("OK  :  %s: expected %s returned %s" % (key, sol, sym))
 
3772
                    print(("OK  :  %s: expected %s returned %s" % (key, sol, sym)))
3772
3773
                else:
3773
 
                    print("FAIL  :  %s: expected %s returned %s" % (key, sol, sym))
 
3774
                    print(("FAIL  :  %s: expected %s returned %s" % (key, sol, sym)))
3774
3775
                    passed = False
3775
3776
            else:
3776
3777
                self.assertEqual(sol, sym, " %s: expected %s returned %s" % (key, sol, sym))
3978
3979
        for wf in [wf for wf in sum([d.get('wavefunctions') for d in \
3979
3980
                                            matrix_elements[0].get('diagrams')\
3980
3981
                                            [1:]], []) if not wf.get('mothers')]:
3981
 
            old_wf = list(filter(lambda w: w.get('number_external') == \
3982
 
                            wf.get('number_external') and not w.get('mothers'),\
3983
 
                            matrix_elements[0].get('diagrams')[0].\
3984
 
                            get('wavefunctions')))[0]
 
3982
            old_wf = list([w for w in matrix_elements[0].get('diagrams')[0].\
 
3983
                            get('wavefunctions') if w.get('number_external') == \
 
3984
                            wf.get('number_external') and not w.get('mothers')])[0]
3985
3985
            self.assertEqual(wf.get('particle'), old_wf.get('particle'))
3986
3986
            self.assert_(wf.get_with_flow('state') != old_wf.get_with_flow('state'))
3987
3987
 
4044
4044
        for wf in [wf for wf in sum([d.get('wavefunctions') for d in \
4045
4045
                                            matrix_elements[0].get('diagrams')\
4046
4046
                                            [1:]], []) if not wf.get('mothers')]:
4047
 
            old_wf = list(filter(lambda w: w.get('number_external') == \
4048
 
                            wf.get('number_external') and not w.get('mothers'),\
4049
 
                            matrix_elements[0].get('diagrams')[0].\
4050
 
                            get('wavefunctions')))[0]
 
4047
            old_wf = list([w for w in matrix_elements[0].get('diagrams')[0].\
 
4048
                            get('wavefunctions') if w.get('number_external') == \
 
4049
                            wf.get('number_external') and not w.get('mothers')])[0]
4051
4050
            self.assertEqual(wf.get('particle'), old_wf.get('particle'))
4052
4051
            self.assert_(wf.get_with_flow('state') != old_wf.get_with_flow('state'))
4053
4052