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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
__authors__ = "Martin Sandve Alnæs"
__date__ = "2008-08-20 -- 2012-11-30"
import pytest
from utils import LagrangeElement
from ufl import Coefficient, FunctionSpace, Mesh, conditional, eq, ge, gt, le, lt, ne, triangle
from ufl.classes import EQ, GE, GT, LE, LT, NE
@pytest.fixture
def f():
element = LagrangeElement(triangle, 1)
domain = Mesh(LagrangeElement(triangle, 1, (2,)))
space = FunctionSpace(domain, element)
return Coefficient(space)
@pytest.fixture
def g():
element = LagrangeElement(triangle, 1)
domain = Mesh(LagrangeElement(triangle, 1, (2,)))
space = FunctionSpace(domain, element)
return Coefficient(space)
def test_conditional_does_not_allow_bool_condition(f, g):
# The reason for this test is that it protects from the case
# conditional(a == b, t, f) in which a == b means comparing representations
with pytest.raises(BaseException):
conditional(True, 1, 0)
def test_eq_oper_produces_bool(f, g):
expr1 = f == f
expr2 = f == g
assert isinstance(expr1, bool)
assert isinstance(expr2, bool)
assert expr1
assert not expr2
def test_eq_produces_ufl_expr(f, g):
expr1 = eq(f, g)
expr2 = eq(f, f)
expr3 = f == g
expr4 = f == f
# Correct types:
assert isinstance(expr1, EQ)
assert isinstance(expr2, EQ)
assert isinstance(expr3, bool)
assert isinstance(expr4, bool)
# Comparing representations correctly:
assert bool(expr1 == eq(f, g))
assert bool(expr1 != eq(g, g))
assert bool(expr2 == eq(f, f))
assert bool(expr2 != eq(g, f))
# Bool evaluation yields actual bools:
assert isinstance(bool(expr1), bool)
assert isinstance(bool(expr2), bool)
assert not expr3
assert expr4
# Allow use in boolean python expression context:
# NB! This means comparing representations! Required by dict and set.
assert not bool(expr1)
assert bool(expr2)
assert not bool(expr3)
assert bool(expr4)
def test_ne_produces_ufl_expr(f, g):
expr1 = ne(f, g)
expr2 = ne(f, f)
expr3 = f != g
expr4 = f != f
# Correct types:
assert isinstance(expr1, NE)
assert isinstance(expr2, NE)
assert isinstance(expr3, bool)
assert isinstance(expr4, bool)
# Comparing representations correctly:
assert bool(expr1 == ne(f, g))
assert bool(expr1 != ne(g, g))
assert bool(expr2 == ne(f, f))
assert bool(expr2 != ne(g, f))
assert not bool(expr2 == expr3)
# Bool evaluation yields actual bools:
assert isinstance(bool(expr1), bool)
assert isinstance(bool(expr2), bool)
# Allow use in boolean python expression context:
# NB! This means the opposite of ==, i.e. comparing representations!
assert bool(expr1)
assert not bool(expr2)
assert bool(expr1)
assert not bool(expr2)
def test_lt_produces_ufl_expr(f, g):
expr1 = lt(f, g)
expr2 = f < g
# Correct types (no bools here!):
assert isinstance(expr1, LT)
assert isinstance(expr2, LT)
# Representations are the same:
assert bool(expr1 == expr2)
# Protection from misuse in boolean python expression context:
with pytest.raises(BaseException):
bool(expr1)
def test_gt_produces_ufl_expr(f, g):
expr1 = gt(f, g)
expr2 = f > g
# Correct types (no bools here!):
assert isinstance(expr1, GT)
assert isinstance(expr2, GT)
# Representations are the same:
assert bool(expr1 == expr2)
# Protection from misuse in boolean python expression context:
with pytest.raises(BaseException):
bool(expr1)
def test_le_produces_ufl_expr(f, g):
expr1 = le(f, g)
expr2 = f <= g
# Correct types (no bools here!):
assert isinstance(expr1, LE)
assert isinstance(expr2, LE)
# Representations are the same:
assert bool(expr1 == expr2)
# Protection from misuse in boolean python expression context:
with pytest.raises(BaseException):
bool(expr1)
def test_ge_produces_ufl_expr(f, g):
expr1 = ge(f, g)
expr2 = f >= g
# Correct types (no bools here!):
assert isinstance(expr1, GE)
assert isinstance(expr2, GE)
# Representations are the same:
assert bool(expr1 == expr2)
# Protection from misuse in boolean python expression context:
with pytest.raises(BaseException):
bool(expr1)
|
