edx_test_count.py

"""Grab the test count and grader setup from the EdX Python exercises."""

import os
import sys

mod = __import__(sys.argv[1][:-3])

def __fn__(*args, **kwargs):
    print('Number of tests:', len(mod.grader.tests()))
    print()
    print(sys.argv[1] + ':')
    with open(os.path.join(os.getcwd(), sys.argv[1]), 'r') as f:
        lines = ['   ' + line.rstrip() for line in f]
    for line in lines[:-1]:
        print(line)
    return lines[-1]  # Return the final line so function based executions don't print None

if hasattr(mod, 'fn'):
    globals()[mod.fn] = __fn__
else:
    print(func())