Update tests.py

Added a test for Comparator