Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 1 | import pytest |
Wenzel Jakob | 1ffce74 | 2016-08-25 01:43:33 +0200 | [diff] [blame] | 2 | import pybind11_tests |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 3 | from pybind11_tests import ConstructorStats |
| 4 | |
| 5 | |
| 6 | def test_override(capture, msg): |
| 7 | from pybind11_tests import (ExampleVirt, runExampleVirt, runExampleVirtVirtual, |
| 8 | runExampleVirtBool) |
| 9 | |
| 10 | class ExtendedExampleVirt(ExampleVirt): |
| 11 | def __init__(self, state): |
| 12 | super(ExtendedExampleVirt, self).__init__(state + 1) |
| 13 | self.data = "Hello world" |
| 14 | |
| 15 | def run(self, value): |
| 16 | print('ExtendedExampleVirt::run(%i), calling parent..' % value) |
| 17 | return super(ExtendedExampleVirt, self).run(value + 1) |
| 18 | |
| 19 | def run_bool(self): |
| 20 | print('ExtendedExampleVirt::run_bool()') |
| 21 | return False |
| 22 | |
Jason Rhinelander | 7dfb932 | 2016-09-08 14:49:43 -0400 | [diff] [blame] | 23 | def get_string1(self): |
| 24 | return "override1" |
| 25 | |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 26 | def pure_virtual(self): |
| 27 | print('ExtendedExampleVirt::pure_virtual(): %s' % self.data) |
| 28 | |
Jason Rhinelander | 7dfb932 | 2016-09-08 14:49:43 -0400 | [diff] [blame] | 29 | class ExtendedExampleVirt2(ExtendedExampleVirt): |
| 30 | def __init__(self, state): |
| 31 | super(ExtendedExampleVirt2, self).__init__(state + 1) |
| 32 | |
| 33 | def get_string2(self): |
| 34 | return "override2" |
| 35 | |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 36 | ex12 = ExampleVirt(10) |
| 37 | with capture: |
| 38 | assert runExampleVirt(ex12, 20) == 30 |
Dean Moldovan | 76e993a | 2016-12-13 00:59:28 +0100 | [diff] [blame] | 39 | assert capture == """ |
| 40 | Original implementation of ExampleVirt::run(state=10, value=20, str1=default1, str2=default2) |
| 41 | """ # noqa: E501 line too long |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 42 | |
| 43 | with pytest.raises(RuntimeError) as excinfo: |
| 44 | runExampleVirtVirtual(ex12) |
| 45 | assert msg(excinfo.value) == 'Tried to call pure virtual function "ExampleVirt::pure_virtual"' |
| 46 | |
| 47 | ex12p = ExtendedExampleVirt(10) |
| 48 | with capture: |
| 49 | assert runExampleVirt(ex12p, 20) == 32 |
| 50 | assert capture == """ |
| 51 | ExtendedExampleVirt::run(20), calling parent.. |
Jason Rhinelander | 7dfb932 | 2016-09-08 14:49:43 -0400 | [diff] [blame] | 52 | Original implementation of ExampleVirt::run(state=11, value=21, str1=override1, str2=default2) |
Dean Moldovan | 76e993a | 2016-12-13 00:59:28 +0100 | [diff] [blame] | 53 | """ # noqa: E501 line too long |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 54 | with capture: |
| 55 | assert runExampleVirtBool(ex12p) is False |
| 56 | assert capture == "ExtendedExampleVirt::run_bool()" |
| 57 | with capture: |
| 58 | runExampleVirtVirtual(ex12p) |
| 59 | assert capture == "ExtendedExampleVirt::pure_virtual(): Hello world" |
| 60 | |
Jason Rhinelander | 7dfb932 | 2016-09-08 14:49:43 -0400 | [diff] [blame] | 61 | ex12p2 = ExtendedExampleVirt2(15) |
| 62 | with capture: |
| 63 | assert runExampleVirt(ex12p2, 50) == 68 |
| 64 | assert capture == """ |
| 65 | ExtendedExampleVirt::run(50), calling parent.. |
| 66 | Original implementation of ExampleVirt::run(state=17, value=51, str1=override1, str2=override2) |
Dean Moldovan | 76e993a | 2016-12-13 00:59:28 +0100 | [diff] [blame] | 67 | """ # noqa: E501 line too long |
Jason Rhinelander | 7dfb932 | 2016-09-08 14:49:43 -0400 | [diff] [blame] | 68 | |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 69 | cstats = ConstructorStats.get(ExampleVirt) |
Jason Rhinelander | 7dfb932 | 2016-09-08 14:49:43 -0400 | [diff] [blame] | 70 | assert cstats.alive() == 3 |
| 71 | del ex12, ex12p, ex12p2 |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 72 | assert cstats.alive() == 0 |
Jason Rhinelander | 7dfb932 | 2016-09-08 14:49:43 -0400 | [diff] [blame] | 73 | assert cstats.values() == ['10', '11', '17'] |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 74 | assert cstats.copy_constructions == 0 |
| 75 | assert cstats.move_constructions >= 0 |
| 76 | |
| 77 | |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 78 | def test_inheriting_repeat(): |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 79 | from pybind11_tests import A_Repeat, B_Repeat, C_Repeat, D_Repeat, A_Tpl, B_Tpl, C_Tpl, D_Tpl |
| 80 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 81 | class AR(A_Repeat): |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 82 | def unlucky_number(self): |
| 83 | return 99 |
| 84 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 85 | class AT(A_Tpl): |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 86 | def unlucky_number(self): |
| 87 | return 999 |
| 88 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 89 | obj = AR() |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 90 | assert obj.say_something(3) == "hihihi" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 91 | assert obj.unlucky_number() == 99 |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 92 | assert obj.say_everything() == "hi 99" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 93 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 94 | obj = AT() |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 95 | assert obj.say_something(3) == "hihihi" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 96 | assert obj.unlucky_number() == 999 |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 97 | assert obj.say_everything() == "hi 999" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 98 | |
| 99 | for obj in [B_Repeat(), B_Tpl()]: |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 100 | assert obj.say_something(3) == "B says hi 3 times" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 101 | assert obj.unlucky_number() == 13 |
| 102 | assert obj.lucky_number() == 7.0 |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 103 | assert obj.say_everything() == "B says hi 1 times 13" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 104 | |
| 105 | for obj in [C_Repeat(), C_Tpl()]: |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 106 | assert obj.say_something(3) == "B says hi 3 times" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 107 | assert obj.unlucky_number() == 4444 |
| 108 | assert obj.lucky_number() == 888.0 |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 109 | assert obj.say_everything() == "B says hi 1 times 4444" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 110 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 111 | class CR(C_Repeat): |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 112 | def lucky_number(self): |
| 113 | return C_Repeat.lucky_number(self) + 1.25 |
| 114 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 115 | obj = CR() |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 116 | assert obj.say_something(3) == "B says hi 3 times" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 117 | assert obj.unlucky_number() == 4444 |
| 118 | assert obj.lucky_number() == 889.25 |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 119 | assert obj.say_everything() == "B says hi 1 times 4444" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 120 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 121 | class CT(C_Tpl): |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 122 | pass |
| 123 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 124 | obj = CT() |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 125 | assert obj.say_something(3) == "B says hi 3 times" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 126 | assert obj.unlucky_number() == 4444 |
| 127 | assert obj.lucky_number() == 888.0 |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 128 | assert obj.say_everything() == "B says hi 1 times 4444" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 129 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 130 | class CCR(CR): |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 131 | def lucky_number(self): |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 132 | return CR.lucky_number(self) * 10 |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 133 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 134 | obj = CCR() |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 135 | assert obj.say_something(3) == "B says hi 3 times" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 136 | assert obj.unlucky_number() == 4444 |
| 137 | assert obj.lucky_number() == 8892.5 |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 138 | assert obj.say_everything() == "B says hi 1 times 4444" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 139 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 140 | class CCT(CT): |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 141 | def lucky_number(self): |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 142 | return CT.lucky_number(self) * 1000 |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 143 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 144 | obj = CCT() |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 145 | assert obj.say_something(3) == "B says hi 3 times" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 146 | assert obj.unlucky_number() == 4444 |
| 147 | assert obj.lucky_number() == 888000.0 |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 148 | assert obj.say_everything() == "B says hi 1 times 4444" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 149 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 150 | class DR(D_Repeat): |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 151 | def unlucky_number(self): |
| 152 | return 123 |
| 153 | |
| 154 | def lucky_number(self): |
| 155 | return 42.0 |
| 156 | |
| 157 | for obj in [D_Repeat(), D_Tpl()]: |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 158 | assert obj.say_something(3) == "B says hi 3 times" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 159 | assert obj.unlucky_number() == 4444 |
| 160 | assert obj.lucky_number() == 888.0 |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 161 | assert obj.say_everything() == "B says hi 1 times 4444" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 162 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 163 | obj = DR() |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 164 | assert obj.say_something(3) == "B says hi 3 times" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 165 | assert obj.unlucky_number() == 123 |
| 166 | assert obj.lucky_number() == 42.0 |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 167 | assert obj.say_everything() == "B says hi 1 times 123" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 168 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 169 | class DT(D_Tpl): |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 170 | def say_something(self, times): |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 171 | return "DT says:" + (' quack' * times) |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 172 | |
| 173 | def unlucky_number(self): |
| 174 | return 1234 |
| 175 | |
| 176 | def lucky_number(self): |
| 177 | return -4.25 |
| 178 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 179 | obj = DT() |
| 180 | assert obj.say_something(3) == "DT says: quack quack quack" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 181 | assert obj.unlucky_number() == 1234 |
| 182 | assert obj.lucky_number() == -4.25 |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 183 | assert obj.say_everything() == "DT says: quack 1234" |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 184 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 185 | class DT2(DT): |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 186 | def say_something(self, times): |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 187 | return "DT2: " + ('QUACK' * times) |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 188 | |
| 189 | def unlucky_number(self): |
| 190 | return -3 |
| 191 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 192 | class BT(B_Tpl): |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 193 | def say_something(self, times): |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 194 | return "BT" * times |
| 195 | |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 196 | def unlucky_number(self): |
| 197 | return -7 |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 198 | |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 199 | def lucky_number(self): |
| 200 | return -1.375 |
| 201 | |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 202 | obj = BT() |
| 203 | assert obj.say_something(3) == "BTBTBT" |
Jason Rhinelander | 2097826 | 2016-08-29 18:16:46 -0400 | [diff] [blame] | 204 | assert obj.unlucky_number() == -7 |
| 205 | assert obj.lucky_number() == -1.375 |
Dean Moldovan | bad1740 | 2016-11-20 21:21:54 +0100 | [diff] [blame] | 206 | assert obj.say_everything() == "BT -7" |
| 207 | |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 208 | |
Wenzel Jakob | 1d1f81b | 2016-12-16 15:00:46 +0100 | [diff] [blame] | 209 | # PyPy: Reference count > 1 causes call with noncopyable instance |
| 210 | # to fail in ncv1.print_nc() |
| 211 | @pytest.unsupported_on_pypy |
Wenzel Jakob | 69b6246 | 2016-08-25 02:20:35 +0200 | [diff] [blame] | 212 | @pytest.mark.skipif(not hasattr(pybind11_tests, 'NCVirt'), |
| 213 | reason="NCVirt test broken on ICPC") |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 214 | def test_move_support(): |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 215 | from pybind11_tests import NCVirt, NonCopyable, Movable |
| 216 | |
| 217 | class NCVirtExt(NCVirt): |
| 218 | def get_noncopyable(self, a, b): |
| 219 | # Constructs and returns a new instance: |
Wenzel Jakob | 69b6246 | 2016-08-25 02:20:35 +0200 | [diff] [blame] | 220 | nc = NonCopyable(a * a, b * b) |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 221 | return nc |
| 222 | |
| 223 | def get_movable(self, a, b): |
| 224 | # Return a referenced copy |
| 225 | self.movable = Movable(a, b) |
| 226 | return self.movable |
| 227 | |
| 228 | class NCVirtExt2(NCVirt): |
| 229 | def get_noncopyable(self, a, b): |
| 230 | # Keep a reference: this is going to throw an exception |
| 231 | self.nc = NonCopyable(a, b) |
| 232 | return self.nc |
| 233 | |
| 234 | def get_movable(self, a, b): |
| 235 | # Return a new instance without storing it |
| 236 | return Movable(a, b) |
| 237 | |
| 238 | ncv1 = NCVirtExt() |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 239 | assert ncv1.print_nc(2, 3) == "36" |
| 240 | assert ncv1.print_movable(4, 5) == "9" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 241 | ncv2 = NCVirtExt2() |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 242 | assert ncv2.print_movable(7, 7) == "14" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 243 | # Don't check the exception message here because it differs under debug/non-debug mode |
| 244 | with pytest.raises(RuntimeError): |
| 245 | ncv2.print_nc(9, 9) |
| 246 | |
| 247 | nc_stats = ConstructorStats.get(NonCopyable) |
| 248 | mv_stats = ConstructorStats.get(Movable) |
| 249 | assert nc_stats.alive() == 1 |
| 250 | assert mv_stats.alive() == 1 |
| 251 | del ncv1, ncv2 |
| 252 | assert nc_stats.alive() == 0 |
| 253 | assert mv_stats.alive() == 0 |
| 254 | assert nc_stats.values() == ['4', '9', '9', '9'] |
| 255 | assert mv_stats.values() == ['4', '5', '7', '7'] |
| 256 | assert nc_stats.copy_constructions == 0 |
| 257 | assert mv_stats.copy_constructions == 1 |
| 258 | assert nc_stats.move_constructions >= 0 |
| 259 | assert mv_stats.move_constructions >= 0 |