mirror of
https://github.com/mealie-recipes/mealie.git
synced 2026-02-25 01:03:13 -05:00
310 lines
8.6 KiB
Python
310 lines
8.6 KiB
Python
import pint
|
|
import pytest
|
|
|
|
from mealie.schema.recipe.recipe_ingredient import CreateIngredientUnit
|
|
from mealie.services.parser_services.parser_utils import UnitConverter, UnitNotFound, merge_quantity_and_unit
|
|
from tests.utils import random_string
|
|
|
|
|
|
def test_uc_parse_string():
|
|
uc = UnitConverter()
|
|
parsed = uc.parse("cup")
|
|
|
|
assert isinstance(parsed, pint.Unit)
|
|
assert (str(parsed)) == "cup"
|
|
|
|
|
|
def test_uc_parse_unit():
|
|
uc = UnitConverter()
|
|
parsed = uc.parse(uc.parse("cup"))
|
|
|
|
assert isinstance(parsed, pint.Unit)
|
|
assert (str(parsed)) == "cup"
|
|
|
|
|
|
def test_uc_parse_invalid():
|
|
uc = UnitConverter()
|
|
input_str = random_string()
|
|
parsed = uc.parse(input_str)
|
|
|
|
assert not isinstance(parsed, pint.Unit)
|
|
assert parsed == input_str
|
|
|
|
|
|
def test_uc_parse_invalid_strict():
|
|
uc = UnitConverter()
|
|
input_str = random_string()
|
|
|
|
with pytest.raises(UnitNotFound):
|
|
uc.parse(input_str, strict=True)
|
|
|
|
|
|
@pytest.mark.parametrize("pre_parse_1", [True, False])
|
|
@pytest.mark.parametrize("pre_parse_2", [True, False])
|
|
def test_can_convert(pre_parse_1: bool, pre_parse_2: bool):
|
|
unit_1 = "cup"
|
|
unit_2 = "pint"
|
|
|
|
uc = UnitConverter()
|
|
if pre_parse_1:
|
|
unit_1 = uc.parse(unit_1)
|
|
if pre_parse_2:
|
|
unit_2 = uc.parse(unit_2)
|
|
|
|
assert uc.can_convert(unit_1, unit_2)
|
|
|
|
|
|
@pytest.mark.parametrize("pre_parse_1", [True, False])
|
|
@pytest.mark.parametrize("pre_parse_2", [True, False])
|
|
def test_cannot_convert(pre_parse_1: bool, pre_parse_2: bool):
|
|
unit_1 = "cup"
|
|
unit_2 = "pound"
|
|
|
|
uc = UnitConverter()
|
|
if pre_parse_1:
|
|
unit_1 = uc.parse(unit_1)
|
|
if pre_parse_2:
|
|
unit_2 = uc.parse(unit_2)
|
|
|
|
assert not uc.can_convert(unit_1, unit_2)
|
|
|
|
|
|
def test_cannot_convert_invalid_unit():
|
|
uc = UnitConverter()
|
|
assert not uc.can_convert("cup", random_string())
|
|
assert not uc.can_convert(random_string(), "cup")
|
|
|
|
|
|
def test_can_convert_same_unit():
|
|
uc = UnitConverter()
|
|
assert uc.can_convert("cup", "cup")
|
|
|
|
|
|
def test_can_convert_volume_ounce():
|
|
uc = UnitConverter()
|
|
assert uc.can_convert("ounce", "cup")
|
|
assert uc.can_convert("cup", "ounce")
|
|
|
|
|
|
def test_convert_simple():
|
|
uc = UnitConverter()
|
|
quantity, unit = uc.convert(1, "cup", "pint")
|
|
|
|
assert isinstance(unit, pint.Unit)
|
|
assert str(unit) == "pint"
|
|
assert quantity == 1 / 2
|
|
|
|
|
|
@pytest.mark.parametrize("pre_parse_1", [True, False])
|
|
@pytest.mark.parametrize("pre_parse_2", [True, False])
|
|
def test_convert_pre_parsed(pre_parse_1: bool, pre_parse_2: bool):
|
|
unit_1 = "cup"
|
|
unit_2 = "pint"
|
|
|
|
uc = UnitConverter()
|
|
if pre_parse_1:
|
|
unit_1 = uc.parse(unit_1)
|
|
if pre_parse_2:
|
|
unit_2 = uc.parse(unit_2)
|
|
|
|
quantity, unit = uc.convert(1, unit_1, unit_2)
|
|
assert isinstance(unit, pint.Unit)
|
|
assert str(unit) == "pint"
|
|
assert quantity == 1 / 2
|
|
|
|
|
|
def test_convert_weight():
|
|
uc = UnitConverter()
|
|
quantity, unit = uc.convert(16, "ounce", "pound")
|
|
|
|
assert isinstance(unit, pint.Unit)
|
|
assert str(unit) == "pound"
|
|
assert quantity == 1
|
|
|
|
|
|
def test_convert_zero_quantity():
|
|
uc = UnitConverter()
|
|
quantity, unit = uc.convert(0, "cup", "pint")
|
|
|
|
assert isinstance(unit, pint.Unit)
|
|
assert quantity == 0
|
|
|
|
|
|
def test_convert_invalid_unit():
|
|
uc = UnitConverter()
|
|
|
|
with pytest.raises(UnitNotFound):
|
|
uc.convert(1, "pound", random_string())
|
|
|
|
|
|
def test_convert_incompatible_units():
|
|
uc = UnitConverter()
|
|
|
|
with pytest.raises(pint.errors.DimensionalityError):
|
|
uc.convert(1, "pound", "cup")
|
|
|
|
|
|
def test_convert_volume_ounce():
|
|
uc = UnitConverter()
|
|
quantity, unit = uc.convert(8, "ounce", "cup")
|
|
|
|
assert isinstance(unit, pint.Unit)
|
|
assert str(unit) == "cup"
|
|
assert quantity == 1
|
|
|
|
|
|
def test_merge_same_unit():
|
|
uc = UnitConverter()
|
|
quantity, unit = uc.merge(1, "cup", 2, "cup")
|
|
|
|
assert isinstance(unit, pint.Unit)
|
|
assert str(unit) == "cup"
|
|
assert quantity == 3
|
|
|
|
|
|
@pytest.mark.parametrize("pre_parse_1", [True, False])
|
|
@pytest.mark.parametrize("pre_parse_2", [True, False])
|
|
def test_merge_compatible_units(pre_parse_1: bool, pre_parse_2: bool):
|
|
unit_1 = "cup"
|
|
unit_2 = "pint"
|
|
|
|
uc = UnitConverter()
|
|
if pre_parse_1:
|
|
unit_1 = uc.parse(unit_1)
|
|
if pre_parse_2:
|
|
unit_2 = uc.parse(unit_2)
|
|
|
|
quantity, unit = uc.merge(1, unit_1, 1, unit_2)
|
|
assert isinstance(unit, pint.Unit)
|
|
# 1 cup + 1 pint = 1 cup + 2 cups = 3 cups
|
|
assert quantity == 3
|
|
|
|
|
|
def test_merge_weight_units():
|
|
uc = UnitConverter()
|
|
quantity, unit = uc.merge(8, "ounce", 8, "ounce")
|
|
|
|
assert isinstance(unit, pint.Unit)
|
|
assert str(unit) == "ounce"
|
|
assert quantity == 16
|
|
|
|
|
|
def test_merge_different_weight_units():
|
|
uc = UnitConverter()
|
|
quantity, unit = uc.merge(1, "pound", 8, "ounce")
|
|
|
|
assert isinstance(unit, pint.Unit)
|
|
# 1 pound + 8 ounces = 16 ounces + 8 ounces = 24 ounces
|
|
assert str(unit) == "pound"
|
|
assert quantity == 1.5
|
|
|
|
|
|
def test_merge_zero_quantities():
|
|
uc = UnitConverter()
|
|
quantity, unit = uc.merge(0, "cup", 1, "cup")
|
|
|
|
assert isinstance(unit, pint.Unit)
|
|
assert str(unit) == "cup"
|
|
assert quantity == 1
|
|
|
|
|
|
def test_merge_invalid_unit():
|
|
uc = UnitConverter()
|
|
|
|
with pytest.raises(UnitNotFound):
|
|
uc.merge(1, "pound", 1, random_string())
|
|
|
|
|
|
def test_merge_incompatible_units():
|
|
uc = UnitConverter()
|
|
|
|
with pytest.raises(pint.errors.DimensionalityError):
|
|
uc.merge(1, "pound", 1, "cup")
|
|
|
|
|
|
def test_merge_negative_quantity():
|
|
uc = UnitConverter()
|
|
quantity, unit = uc.merge(-1, "cup", 2, "cup")
|
|
|
|
assert isinstance(unit, pint.Unit)
|
|
assert str(unit) == "cup"
|
|
assert quantity == 1
|
|
|
|
|
|
def test_merge_volume_ounce():
|
|
uc = UnitConverter()
|
|
quantity, unit = uc.merge(4, "ounce", 1, "cup")
|
|
|
|
assert isinstance(unit, pint.Unit)
|
|
assert str(unit) == "fluid_ounce" # converted automatically from ounce
|
|
assert quantity == 12
|
|
|
|
|
|
def test_merge_quantity_and_unit_simple():
|
|
unit_1 = CreateIngredientUnit(name="mealie_cup", standard_quantity=1, standard_unit="cup")
|
|
unit_2 = CreateIngredientUnit(name="mealie_cup", standard_quantity=1, standard_unit="cup")
|
|
|
|
quantity, unit = merge_quantity_and_unit(1, unit_1, 2, unit_2)
|
|
|
|
assert quantity == 3
|
|
assert unit.name == "mealie_cup"
|
|
|
|
|
|
def test_merge_quantity_and_unit_invalid():
|
|
unit_1 = CreateIngredientUnit(name="mealie_cup", standard_quantity=1, standard_unit="cup")
|
|
unit_2 = CreateIngredientUnit(name="mealie_random", standard_quantity=1, standard_unit=random_string())
|
|
|
|
with pytest.raises(UnitNotFound):
|
|
merge_quantity_and_unit(1, unit_1, 1, unit_2)
|
|
|
|
|
|
def test_merge_quantity_and_unit_compatible():
|
|
unit_1 = CreateIngredientUnit(name="mealie_pint", standard_quantity=1, standard_unit="pint")
|
|
unit_2 = CreateIngredientUnit(name="mealie_cup", standard_quantity=1, standard_unit="cup")
|
|
|
|
quantity, unit = merge_quantity_and_unit(1, unit_1, 1, unit_2)
|
|
|
|
# 1 pint + 1 cup = 2 pints + 1 cup = 3 cups, converted to pint = 1.5 pint
|
|
assert quantity == 1.5
|
|
assert unit.name == "mealie_pint"
|
|
|
|
|
|
def test_merge_quantity_and_unit_selects_larger_unit():
|
|
unit_1 = CreateIngredientUnit(name="mealie_pint", standard_quantity=1, standard_unit="pint")
|
|
unit_2 = CreateIngredientUnit(name="mealie_cup", standard_quantity=1, standard_unit="cup")
|
|
|
|
quantity, unit = merge_quantity_and_unit(2, unit_1, 4, unit_2)
|
|
|
|
# 2 pint + 4 cup = 4 cups + 4 cups = 8 cups, should be returned as pint (larger unit)
|
|
assert quantity == 4
|
|
assert unit.name == "mealie_pint"
|
|
|
|
|
|
def test_merge_quantity_and_unit_selects_smaller_unit():
|
|
unit_1 = CreateIngredientUnit(name="mealie_pint", standard_quantity=1, standard_unit="pint")
|
|
unit_2 = CreateIngredientUnit(name="mealie_cup", standard_quantity=1, standard_unit="cup")
|
|
|
|
quantity, unit = merge_quantity_and_unit(0.125, unit_1, 0.5, unit_2)
|
|
|
|
# 0.125 pint + 0.5 cup = 0.25 cup + 0.5 cup = 0.75 cup, should be returned as cup (smaller for < 1)
|
|
assert quantity == 0.75
|
|
assert unit.name == "mealie_cup"
|
|
|
|
|
|
def test_merge_quantity_and_unit_missing_standard_data():
|
|
unit_1 = CreateIngredientUnit(name="mealie_cup", standard_quantity=1, standard_unit="cup")
|
|
unit_2 = CreateIngredientUnit(name="mealie_cup_no_std", standard_quantity=None, standard_unit=None)
|
|
|
|
with pytest.raises(ValueError):
|
|
merge_quantity_and_unit(1, unit_1, 1, unit_2)
|
|
|
|
|
|
def test_merge_quantity_and_unit_volume_ounce():
|
|
unit_1 = CreateIngredientUnit(name="mealie_oz", standard_quantity=1, standard_unit="ounce")
|
|
unit_2 = CreateIngredientUnit(name="mealie_cup", standard_quantity=1, standard_unit="cup")
|
|
|
|
quantity, unit = merge_quantity_and_unit(8, unit_1, 1, unit_2)
|
|
|
|
assert quantity == 2
|
|
assert unit.name == "mealie_cup"
|