lpython/integration_tests/enum_02.py at main · sak-codes/lpython · GitHub

39 lines (31 loc) · 1.27 KB

from enum import Enum

from ltypes import i32, f64

class MolecularMass(Enum):

water: i32 = 18

methane: i32 = 16

ammonia: i32 = 17

oxygen: i32 = 16

class RelativeCharge(Enum):

proton: i32 = 1

electron: i32 = -1

neutron: i32 = 0

class AbsoluteChargeWithoutExp(Enum):

proton: f64 = 1.602 # ignoring 10**(-19) because of printing limitations

electron: f64 = -1.602

neutron: f64 = 0.0

def test_mm():

assert MolecularMass.methane.value == MolecularMass.oxygen.value

assert MolecularMass.ammonia.value - MolecularMass.methane.value == 1

assert MolecularMass.water.value == MolecularMass.oxygen.value + 2

def test_rc():

assert RelativeCharge.proton.value + RelativeCharge.electron.value == RelativeCharge.neutron.value

def charge_ratio(c1: AbsoluteChargeWithoutExp, c2: AbsoluteChargeWithoutExp) -> f64:

return c1.value/c2.value

def test_ac():

print(AbsoluteChargeWithoutExp.proton.name, AbsoluteChargeWithoutExp.proton.value)

print(AbsoluteChargeWithoutExp.electron.name, AbsoluteChargeWithoutExp.electron.value)

print(AbsoluteChargeWithoutExp.neutron.name, AbsoluteChargeWithoutExp.neutron.value)

assert charge_ratio(AbsoluteChargeWithoutExp.proton, AbsoluteChargeWithoutExp.electron) == -1.0

test_mm()

test_rc()

test_ac()

Expand file tree