File: operator_overloading_27.f90

package info (click to toggle)
lfortran 0.61.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 61,892 kB
  • sloc: cpp: 181,767; f90: 92,175; python: 17,616; ansic: 10,170; yacc: 2,377; sh: 1,444; fortran: 892; makefile: 38; javascript: 15
file content (57 lines) | stat: -rw-r--r-- 1,532 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
module module_operator_overloading_27
  implicit none

  type :: result_t
    logical :: passed = .false.
  end type

  interface operator(.check.)
    module procedure check
  end interface

  interface operator(.and.)
    module procedure and_op
  end interface

contains

  elemental function check(actual, expected) result(res)
    integer, intent(in) :: actual, expected
    type(result_t) :: res
    res%passed = (actual == expected)
  end function

  elemental function and_op(lhs, rhs) result(res)
    type(result_t), intent(in) :: lhs, rhs
    type(result_t) :: res
    res%passed = lhs%passed .and. rhs%passed
  end function

end module

program operator_overloading_27
  use module_operator_overloading_27
  implicit none
  type(result_t) :: diagnosis(3)
  type(result_t) :: scalar_res

  ! Test: user-defined .and. on arrays returned by user-defined .check.
  diagnosis = (2 .check. [2,2,2]) .and. ([0,1,2] .check. [0,1,2])
  if (.not. all(diagnosis%passed)) error stop

  ! Test: user-defined .and. on scalars
  scalar_res = (3 .check. 3) .and. (4 .check. 4)
  if (.not. scalar_res%passed) error stop

  ! Test: one side fails
  scalar_res = (3 .check. 3) .and. (4 .check. 5)
  if (scalar_res%passed) error stop

  ! Test: arrays with partial mismatch
  diagnosis = (1 .check. [1,2,1]) .and. ([3,3,3] .check. [3,3,3])
  if (diagnosis(1)%passed .neqv. .true.) error stop
  if (diagnosis(2)%passed .neqv. .false.) error stop
  if (diagnosis(3)%passed .neqv. .true.) error stop

  print *, "All tests passed."
end program