File: mcomputearray.py

package info (click to toggle)
calceph 4.0.4-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 15,944 kB
  • sloc: ansic: 21,414; fortran: 4,054; python: 1,569; sh: 197; makefile: 5
file content (114 lines) | stat: -rw-r--r-- 4,410 bytes parent folder | download | duplicates (2)
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
# /*-----------------------------------------------------------------*/
# /*!
#  \file mcomputearray.py
#  \brief Check the function calceph_compute with vectors.
#
#  \author  M. Gastineau
#           Astronomie et Systemes Dynamiques, IMCCE, CNRS, Observatoire de Paris.
#
#   Copyright, 2019, CNRS
#   email of the author : Mickael.Gastineau@obspm.fr
# */
# /*-----------------------------------------------------------------*/
#
# /*-----------------------------------------------------------------*/
# /* License  of this file :
# This file is "triple-licensed", you have to choose one  of the three licenses
# below to apply on this file.
#
#    CeCILL-C
#    	The CeCILL-C license is close to the GNU LGPL.
#    	( http://www.cecill.info/licences/Licence_CeCILL-C_V1-en.html )
#
# or CeCILL-B
#        The CeCILL-B license is close to the BSD.
#        (http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.txt)
#
# or CeCILL v2.1
#      The CeCILL license is compatible with the GNU GPL.
#      ( http://www.cecill.info/licences/Licence_CeCILL_V2.1-en.html )
#
#
# This library is governed by the CeCILL-C, CeCILL-B or the CeCILL license under
# French law and abiding by the rules of distribution of free software.
# You can  use, modify and/ or redistribute the software under the terms
# of the CeCILL-C,CeCILL-B or CeCILL license as circulated by CEA, CNRS and INRIA
# at the following URL "http://www.cecill.info".
#
# As a counterpart to the access to the source code and  rights to copy,
# modify and redistribute granted by the license, users are provided only
# with a limited warranty  and the software's author,  the holder of the
# economic rights,  and the successive licensors  have only  limited
# liability.
#
# In this respect, the user's attention is drawn to the risks associated
# with loading,  using,  modifying and/or developing or reproducing the
# software by the user in light of its specific status of free software,
# that may mean  that it is complicated to manipulate,  and  that  also
# therefore means  that it is reserved for developers  and  experienced
# professionals having in-depth computer knowledge. Users are therefore
# encouraged to load and test the software's suitability as regards their
# requirements in conditions enabling the security of their systems and/or
# data to be ensured and,  more generally, to use and operate it in the
# same conditions as regards security.
#
# The fact that you are presently reading this means that you have had
# knowledge of the CeCILL-C,CeCILL-B or CeCILL license and that you accept its terms.
# */
# /*-----------------------------------------------------------------*/

# /*-----------------------------------------------------------------*/
# /* main program */
# /*-----------------------------------------------------------------*/
from calcephpy import CalcephBin
import openfiles
import unittest
import numpy


class TestOpen(unittest.TestCase):

    # set different size of arrays
    def test_mcompute(self):
        res = 1
        peph = CalcephBin.open(openfiles.prefixsrc(
            "../../examples/example1.dat"))
        jd0 = range(2442457,  2442457+10)
        dt = range(0, 10)
        target = 1
        center = 12
        PVlist = peph.compute(jd0, dt, target, center)
        PVnumpy = peph.compute(numpy.asarray(jd0, dtype=numpy.float64), numpy.asarray(
            dt, dtype=numpy.float64), target, center)

        if len(PVlist) != 6:
            len(PVarray)
            raise RuntimeError("bad size array list")

        if len(PVlist[0]) != 10:
            len(PVarray)
            raise RuntimeError("bad size array list ")

        if len(PVnumpy) != 6:
            len(PVarray)
            raise RuntimeError("bad size array numpy")

        if len(PVnumpy[0]) != 10:
            len(PVarray)
            raise RuntimeError("bad size array numpy")

        for j in range(len(jd0)):
            PV = peph.compute(jd0[j], dt[j], target, center)
            for k in range(6):
                if PV[k] != PVlist[k][j]:
                    print("j %d" % j)
                    print("expected :", PV)
                    print("computed :", PVlist[:][j])
                if PV[k] != PVnumpy[k][j]:
                    print("j %d" % j)
                    print("expected :", PV)
                    print("computed :", PVlist[:][j])


if __name__ == '__main__':
    unittest.main()