mirror of
https://github.com/galaxyproject/galaxy.git
synced 2026-09-24 16:30:27 +08:00
Add utility function for enum member values + test
This commit is contained in:
@@ -1905,3 +1905,11 @@ class StructuredExecutionTimer:
|
||||
@property
|
||||
def elapsed(self):
|
||||
return time.time() - self.begin
|
||||
|
||||
|
||||
def enum_values(enum_class):
|
||||
"""
|
||||
Return a list of member values of enumeration enum_class.
|
||||
Values are in member definition order.
|
||||
"""
|
||||
return [item.value for key, item in enum_class.__members__.items()]
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
import errno
|
||||
import os
|
||||
import tempfile
|
||||
from enum import Enum
|
||||
from io import StringIO
|
||||
from typing import Dict
|
||||
|
||||
@@ -139,3 +140,12 @@ def test_listify() -> None:
|
||||
assert util.listify(d) == [d]
|
||||
o = object()
|
||||
assert util.listify(o) == [o]
|
||||
|
||||
|
||||
def test_enum_values():
|
||||
class Stuff(str, Enum):
|
||||
A = "a"
|
||||
C = "c"
|
||||
B = "b"
|
||||
|
||||
assert util.enum_values(Stuff) == ["a", "c", "b"]
|
||||
|
||||
Reference in New Issue
Block a user