Adds tests for definition_extension method