def get_version(): """Get version information""" airflow_version = airflow.__version__ git_version = get_airflow_git_version() version_info = VersionInfo(version=airflow_version, git_version=git_version) return version_info_schema.dump(version_info)
def test_serialize(self, git_commit): version_info = VersionInfo("VERSION", git_commit) current_data = version_info_schema.dump(version_info) expected_result = {'version': 'VERSION', 'git_version': git_commit} assert expected_result == current_data