def test_circle_with_mocked_rectangle(mocked_rectangle):
mocked_rectangle.return_value = 12
result = calculate_area('SQUARE', 2)
assert result == 12
mocked_rectangle.assert_called()
def test_square():
result = calculate_area('SQUARE', 2)
assert result == 4
def test_circle_with_proper_pi():
result = calculate_area('CIRCLE', 2)
assert result == 12.56636
def test_circle_with_mocked_pi():
result = calculate_area('CIRCLE', 2)
assert result == 12
def test_rectangle():
result = calculate_area('RECTANGLE', 2, 3)
assert result == 6