def test_get_water_level_badge_class_returns_correct_badge(
self, garden_factory, value, klass):
formatter = GardenFormatter(garden_factory.build(water_level=value))
ret_val = formatter.get_water_level_badge_class()
assert ret_val == klass
def test_get_water_level_element_return_value_contains_water_level_display_and_badge_classes(
self, garden_factory):
formatter = GardenFormatter(garden_factory.build())
ret1 = 'ret1'
formatter.get_water_level_badge_class = Mock(return_value=ret1)
ret_val = formatter.get_water_level_element()
assert formatter.instance.get_water_level_display() in ret_val
assert formatter.get_water_level_badge_class.return_value in ret_val