diff --git a/styles/tests_extractor.d b/styles/tests_extractor.d index 7213c32..0bcfdc4 100644 --- a/styles/tests_extractor.d +++ b/styles/tests_extractor.d @@ -54,11 +54,10 @@ class TestVisitor : ASTVisitor override void visit(const Declaration decl) { - if (decl.unittest_ !is null) - { - if (hasDdocHeader(sourceCode, decl)) - print(decl.unittest_); - } + if (decl.unittest_ !is null && hasDdocHeader(sourceCode, decl)) + print(decl.unittest_); + + decl.accept(this); } private: