ITQC/pydase
9
0
Fork 0
mirror of https://github.com/tiqi-group/pydase.git synced 2025-12-29 17:11:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

updating pydase.units

Browse Source
This commit is contained in:
Mose Müller
2023-09-19 14:04:21 +02:00
parent fe5b6c591d
commit c182e11527
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/pydase/units.py
View File

@@ -1,13 +1,11 @@
from typing import TypedDict, Union
import pint
from pint import Quantity
units: pint.UnitRegistry = pint.UnitRegistry()
units = pint.UnitRegistry()
units.default_format = "~P" # pretty and short format
Quantity = pint.Quantity
Unit = units.Unit
class QuantityDict(TypedDict):
magnitude: Union[int, float]
@@ -48,9 +46,9 @@ def convert_to_quantity(
"""
if isinstance(value, (int, float)):
quantity = float(value) * Unit(unit)
quantity = float(value) * units(unit)
elif isinstance(value, dict):
quantity = float(value["magnitude"]) * Unit(value["unit"])
quantity = float(value["magnitude"]) * units(value["unit"])
else:
quantity = value
return quantity # type: ignore