added test in tests.py. it passes