Dean Moldovan | 382db5b | 2016-08-13 00:37:50 +0200 | [diff] [blame] | 1 | import pytest |
| 2 | |
| 3 | |
| 4 | def test_inheritance(msg): |
Dean Moldovan | 08cbe8d | 2017-02-15 21:10:25 +0100 | [diff] [blame] | 5 | from pybind11_tests import Pet, Dog, Rabbit, Hamster, Chimera, dog_bark, pet_name_species |
Dean Moldovan | 382db5b | 2016-08-13 00:37:50 +0200 | [diff] [blame] | 6 | |
| 7 | roger = Rabbit('Rabbit') |
| 8 | assert roger.name() + " is a " + roger.species() == "Rabbit is a parrot" |
| 9 | assert pet_name_species(roger) == "Rabbit is a parrot" |
| 10 | |
| 11 | polly = Pet('Polly', 'parrot') |
| 12 | assert polly.name() + " is a " + polly.species() == "Polly is a parrot" |
| 13 | assert pet_name_species(polly) == "Polly is a parrot" |
| 14 | |
| 15 | molly = Dog('Molly') |
| 16 | assert molly.name() + " is a " + molly.species() == "Molly is a dog" |
| 17 | assert pet_name_species(molly) == "Molly is a dog" |
| 18 | |
Jason Rhinelander | 6b52c83 | 2016-09-06 12:27:00 -0400 | [diff] [blame] | 19 | fred = Hamster('Fred') |
| 20 | assert fred.name() + " is a " + fred.species() == "Fred is a rodent" |
| 21 | |
Dean Moldovan | 382db5b | 2016-08-13 00:37:50 +0200 | [diff] [blame] | 22 | assert dog_bark(molly) == "Woof!" |
| 23 | |
| 24 | with pytest.raises(TypeError) as excinfo: |
| 25 | dog_bark(polly) |
| 26 | assert msg(excinfo.value) == """ |
Wenzel Jakob | e99ebae | 2016-09-12 11:44:37 +0900 | [diff] [blame] | 27 | dog_bark(): incompatible function arguments. The following argument types are supported: |
Dean Moldovan | 382db5b | 2016-08-13 00:37:50 +0200 | [diff] [blame] | 28 | 1. (arg0: m.Dog) -> str |
Wenzel Jakob | e99ebae | 2016-09-12 11:44:37 +0900 | [diff] [blame] | 29 | |
| 30 | Invoked with: <m.Pet object at 0> |
Dean Moldovan | 382db5b | 2016-08-13 00:37:50 +0200 | [diff] [blame] | 31 | """ |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 32 | |
Dean Moldovan | 08cbe8d | 2017-02-15 21:10:25 +0100 | [diff] [blame] | 33 | with pytest.raises(TypeError) as excinfo: |
| 34 | Chimera("lion", "goat") |
| 35 | assert "No constructor defined!" in str(excinfo.value) |
| 36 | |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 37 | |
| 38 | def test_automatic_upcasting(): |
Jason Rhinelander | 0e48977 | 2016-09-11 18:41:28 -0400 | [diff] [blame] | 39 | from pybind11_tests import return_class_1, return_class_2, return_class_n, return_none |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 40 | |
| 41 | assert type(return_class_1()).__name__ == "DerivedClass1" |
| 42 | assert type(return_class_2()).__name__ == "DerivedClass2" |
| 43 | assert type(return_none()).__name__ == "NoneType" |
Pim Schellart | cc88aae | 2017-01-31 10:52:11 -0500 | [diff] [blame] | 44 | # Repeat these a few times in a random order to ensure no invalid caching |
| 45 | # is applied |
Jason Rhinelander | 0e48977 | 2016-09-11 18:41:28 -0400 | [diff] [blame] | 46 | assert type(return_class_n(1)).__name__ == "DerivedClass1" |
| 47 | assert type(return_class_n(2)).__name__ == "DerivedClass2" |
| 48 | assert type(return_class_n(0)).__name__ == "BaseClass" |
| 49 | assert type(return_class_n(2)).__name__ == "DerivedClass2" |
| 50 | assert type(return_class_n(2)).__name__ == "DerivedClass2" |
| 51 | assert type(return_class_n(0)).__name__ == "BaseClass" |
| 52 | assert type(return_class_n(1)).__name__ == "DerivedClass1" |
Dean Moldovan | b4498ef | 2016-10-23 14:50:08 +0200 | [diff] [blame] | 53 | |
| 54 | |
| 55 | def test_isinstance(): |
| 56 | from pybind11_tests import test_isinstance, Pet, Dog |
| 57 | |
| 58 | objects = [tuple(), dict(), Pet("Polly", "parrot")] + [Dog("Molly")] * 4 |
| 59 | expected = (True, True, True, True, True, False, False) |
| 60 | assert test_isinstance(objects) == expected |
Pim Schellart | cc88aae | 2017-01-31 10:52:11 -0500 | [diff] [blame] | 61 | |
| 62 | |
| 63 | def test_holder(): |
| 64 | from pybind11_tests import test_mismatched_holder_type_1, test_mismatched_holder_type_2 |
| 65 | |
| 66 | with pytest.raises(RuntimeError) as excinfo: |
| 67 | test_mismatched_holder_type_1() |
| 68 | |
| 69 | assert str(excinfo.value) == ("generic_type: type \"MismatchDerived1\" does not have " |
| 70 | "a non-default holder type while its base " |
| 71 | "\"MismatchBase1\" does") |
| 72 | |
| 73 | with pytest.raises(RuntimeError) as excinfo: |
| 74 | test_mismatched_holder_type_2() |
| 75 | |
| 76 | assert str(excinfo.value) == ("generic_type: type \"MismatchDerived2\" has a " |
| 77 | "non-default holder type while its base " |
| 78 | "\"MismatchBase2\" does not") |