Create test_category.py