ENDFtk/python/test/Test_ENDFtk_MF14_TabulatedDistribution.py at cccab26bb729b3160487faf0fd6b78ef004344ed · njoy/ENDFtk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
# standard imports
import unittest

# third party imports

# local imports
from ENDFtk.MF14 import TabulatedDistribution

class Test_ENDFtk_MF4_TabulatedDistribution( unittest.TestCase ) :
    """Unit test for the TabulatedDistribution class."""

    chunk = ( ' 0.000000+0 1.000000-5          0          0          1          3922814  2     \n'
              '          3          2                                            922814  2     \n'
              '-1.000000+0 0.000000+0 0.000000+0 1.000000+0 1.000000+0 0.000000+0922814  2     \n' )

    invalid = ( ' 0.000000+0 1.000000-5          0          0          2          3922814  2     \n'
                '          3          2                                            922814  2     \n'
                '-1.000000+0 0.000000+0 0.000000+0 1.000000+0 1.000000+0 0.000000+0922814  2     \n' )

    def test_component( self ) :

        def verify_chunk( self, chunk ) :

            # verify content
            self.assertAlmostEqual( 1e-5, chunk.E )
            self.assertAlmostEqual( 1e-5, chunk.incident_energy )
            self.assertEqual( 3, chunk.NP )
            self.assertEqual( 1, chunk.NR )
            self.assertEqual( 1, len( chunk.boundaries ) )
            self.assertEqual( 1, len( chunk.interpolants ) )
            self.assertEqual( 3, chunk.boundaries[0] )
            self.assertEqual( 2, chunk.interpolants[0] )
            self.assertEqual( 3, len( chunk.cosines ) )
            self.assertAlmostEqual( -1.0 , chunk.MU[0] )
            self.assertAlmostEqual( 0.0, chunk.MU[1] )
            self.assertAlmostEqual( 1.0, chunk.MU[2] )
            self.assertAlmostEqual( -1.0 , chunk.cosines[0] )
            self.assertAlmostEqual( 0.0, chunk.cosines[1] )
            self.assertAlmostEqual( 1.0, chunk.cosines[2] )
            self.assertEqual( 3, len( chunk.probabilities ) )
            self.assertAlmostEqual( 0.0 , chunk.F[0] )
            self.assertAlmostEqual( 1.0, chunk.F[1] )
            self.assertAlmostEqual( 0.0, chunk.F[2] )
            self.assertAlmostEqual( 0.0 , chunk.probabilities[0] )
            self.assertAlmostEqual( 1.0, chunk.probabilities[1] )
            self.assertAlmostEqual( 0.0, chunk.probabilities[2] )

            self.assertEqual( 3, chunk.NC )

            # verify string
            self.assertEqual( self.chunk, chunk.to_string( 9228, 14, 2 ) )

        # the data is given explicitly
        chunk = TabulatedDistribution( energy = 1e-5,
                                       boundaries = [ 3 ], interpolants = [ 2 ],
                                       cosines = [ -1.0, 0.0, 1.0 ], probabilities = [ 0.0, 1.0, 0.0 ] )

        verify_chunk( self, chunk )

        # the data is read from a string
        chunk = TabulatedDistribution.from_string( self.chunk, 9228, 14, 2 )

        verify_chunk( self, chunk )

        # the data is copied
        copy = TabulatedDistribution( chunk )

        verify_chunk( self, copy )

    def test_failures( self ) :

        print( '\n' )

        # wrong boundaries
        with self.assertRaises( Exception ) :

            chunk = TabulatedDistribution(
                        energy = 1e-5,
                        boundaries = [ 4 ], interpolants = [ 2 ],
                        cosines = [ -1.0, 0.0, 1.0 ], probabilities = [ 0.0, 1.0, 0.0 ] )

        with self.assertRaises( Exception ) :

            chunk = TabulatedDistribution.from_string( self.invalid, 9228, 14, 2 )

if __name__ == '__main__' :

    unittest.main()