File: complex_30.f90

package info (click to toggle)
lfortran 0.59.0-3
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 56,736 kB
  • sloc: cpp: 168,052; f90: 74,272; python: 17,537; ansic: 7,705; yacc: 2,345; sh: 1,334; fortran: 895; makefile: 37; javascript: 15
file content (44 lines) | stat: -rw-r--r-- 1,617 bytes parent folder | download | duplicates (3)
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
program complex_ops_01
    ! Test that complex arithmetic works correctly
    ! This also serves to verify inline complex ops when implemented
    implicit none
    complex :: a, b, c, d
    real :: eps

    eps = 1.0e-6

    ! Test complex multiplication: (1+2i) * (3+4i) = (1*3-2*4) + (1*4+2*3)i = -5 + 10i
    a = (1.0, 2.0)
    b = (3.0, 4.0)
    c = a * b
    if (abs(real(c) - (-5.0)) > eps) error stop "mul real failed"
    if (abs(aimag(c) - 10.0) > eps) error stop "mul imag failed"

    ! Test complex addition: (1+2i) + (3+4i) = 4 + 6i
    c = a + b
    if (abs(real(c) - 4.0) > eps) error stop "add real failed"
    if (abs(aimag(c) - 6.0) > eps) error stop "add imag failed"

    ! Test complex subtraction: (1+2i) - (3+4i) = -2 - 2i
    c = a - b
    if (abs(real(c) - (-2.0)) > eps) error stop "sub real failed"
    if (abs(aimag(c) - (-2.0)) > eps) error stop "sub imag failed"

    ! Test complex division: (1+2i) / (3+4i) = (1*3+2*4)/(3^2+4^2) + (2*3-1*4)/(3^2+4^2)i
    ! = 11/25 + 2/25 i = 0.44 + 0.08i
    c = a / b
    if (abs(real(c) - 0.44) > eps) error stop "div real failed"
    if (abs(aimag(c) - 0.08) > eps) error stop "div imag failed"

    ! Test compound expression
    d = (a * b + a) / b - a
    ! = ((-5+10i) + (1+2i)) / (3+4i) - (1+2i)
    ! = (-4+12i) / (3+4i) - (1+2i)
    ! = ((-4*3+12*4)/25 + (12*3-(-4)*4)/25 i) - (1+2i)
    ! = (36/25 + 52/25 i) - (1+2i)
    ! = (1.44 - 1) + (2.08 - 2)i = 0.44 + 0.08i
    if (abs(real(d) - 0.44) > eps) error stop "compound real failed"
    if (abs(aimag(d) - 0.08) > eps) error stop "compound imag failed"

    print *, "PASS"
end program