pqc/test/test_duplicate_consistency.py

"""
Checks that files duplicated across schemes/implementations are consistent.
"""

import os
import pqclean
import helpers
import unittest
import yaml

helpers.skip_windows()
def test_duplicate_consistency():
    for scheme in pqclean.Scheme.all_schemes():
        for implementation in scheme.implementations:
            if helpers.permit_test('duplicate_consistency', implementation): 
                if os.path.isfile(os.path.join('duplicate_consistency', '{}_{}.yml'.format(scheme.name, implementation.name))):
                    metafile = os.path.join('duplicate_consistency', '{}_{}.yml'.format(implementation.scheme.name, implementation.name))
                    with open(metafile, encoding='utf-8') as f:
                        metadata = yaml.safe_load(f.read())
                        for group in metadata['consistency_checks']:
                            source = pqclean.Implementation.by_name(group['source']['scheme'], group['source']['implementation'])
                            for file in group['files']:
                                yield check_duplicate_consistency, implementation, source, file

def file_get_contents(filename):
    with open(filename) as f:
        return f.read()

def check_duplicate_consistency(implementation, source, file):
    transformed_src = helpers.run_subprocess(
        ['sed', '-e', 's/{}/{}/g'.format(source.namespace_prefix(), implementation.namespace_prefix()), os.path.join(source.path(), file)]
    )
    this_src = file_get_contents(os.path.join(implementation.path(), file))
    print(os.path.join(implementation.path(), file))
    print(this_src)
    assert(transformed_src == this_src)

if __name__ == '__main__':
    try:
        import nose2
        nose2.main()
    except ImportError:
        import nose
        nose.runmodule()