2022-03-18 22:36:27 +00:00

26 lines
622 B
Python

import pytest
from tests.output_betterproto.nestedtwice import (
Test,
TestTop,
TestTopMiddle,
TestTopMiddleBottom,
TestTopMiddleEnumBottom,
TestTopMiddleTopMiddleBottom,
)
@pytest.mark.parametrize(
("cls", "expected_comment"),
[
(Test, "Test doc."),
(TestTopMiddleEnumBottom, "EnumBottom doc."),
(TestTop, "Top doc."),
(TestTopMiddle, "Middle doc."),
(TestTopMiddleTopMiddleBottom, "TopMiddleBottom doc."),
(TestTopMiddleBottom, "Bottom doc."),
],
)
def test_comment(cls, expected_comment):
assert cls.__doc__ == expected_comment