Update tests.py

John added EQ classes.