File: test_units.py

package info (click to toggle)
einsteinpy 0.4.0-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 40,712 kB
  • sloc: python: 8,196; makefile: 146
file content (25 lines) | stat: -rw-r--r-- 546 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
import astropy.units as u
import numpy as np

from numpy.testing import assert_allclose

from einsteinpy.units import primitive


def test_primitive():
    """
    Tests, if quantities are safely being converted into ``numpy.float64`` values

    """
    M = 10 * u.kg
    a = 0.29 * u.one
    Q = 1e3 * u.C
    t = 1e3 * u.s
    r = 1e10 * u.m
    theta = np.pi / 8 * u.rad
    temp = 1e5

    values = [10., 0.29, 1e3, 1e3, 1e10, np.pi / 8, 1e5]
    values_calc = primitive(M, a, Q, t, r, theta, temp)

    assert_allclose(values, values_calc)