Create test_class.py

A work in progress.