def test_capitalize_months_no_months(self):
"""Ensures correct output from _capitalize_months.
In this case, original string contains no months.
"""
this_new_string = storm_events_io._capitalize_months(
STRING_WITH_NO_MONTHS)
self.assertTrue(this_new_string == STRING_WITH_NO_MONTHS)
def test_capitalize_months_1month_correct_caps(self):
"""Ensures correct output from _capitalize_months.
In this case, original string contains 1 month with the correct caps.
"""
this_new_string = storm_events_io._capitalize_months(
STRING_WITH_1MONTH_CORRECT_CAPS)
self.assertTrue(this_new_string == STRING_WITH_1MONTH_CORRECT_CAPS)
def test_capitalize_months_many_months(self):
"""Ensures correct output from _capitalize_months.
In this case, original string contains many months with the wrONG
CAPitalization.
"""
this_new_string = storm_events_io._capitalize_months(
STRING_WITH_MANY_MONTHS)
self.assertTrue(this_new_string == STRING_WITH_MANY_MONTHS_FIXED)