219eb09e59
The loading protocol mandates that the the module we are going to import needs to be already in sys.modules before its code is executed, so to prevent unbounded recursions and multiple loading. Loading a module from file exits early if the module is already in sys.modules
158 lines
4.4 KiB
Python
158 lines
4.4 KiB
Python
# Copyright 2013-2021 Lawrence Livermore National Security, LLC and other
|
|
# Spack Project Developers. See the top-level COPYRIGHT file for details.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
|
|
import pytest
|
|
|
|
import os.path
|
|
import sys
|
|
from datetime import datetime, timedelta
|
|
|
|
import llnl.util.lang
|
|
from llnl.util.lang import pretty_date, match_predicate
|
|
|
|
|
|
@pytest.fixture()
|
|
def now():
|
|
return datetime.now()
|
|
|
|
|
|
@pytest.fixture()
|
|
def module_path(tmpdir):
|
|
m = tmpdir.join('foo.py')
|
|
content = """
|
|
import os.path
|
|
|
|
value = 1
|
|
path = os.path.join('/usr', 'bin')
|
|
"""
|
|
m.write(content)
|
|
|
|
yield str(m)
|
|
|
|
# Don't leave garbage in the module system
|
|
if 'foo' in sys.modules:
|
|
del sys.modules['foo']
|
|
|
|
|
|
def test_pretty_date():
|
|
"""Make sure pretty_date prints the right dates."""
|
|
now = datetime.now()
|
|
|
|
just_now = now - timedelta(seconds=5)
|
|
assert pretty_date(just_now, now) == "just now"
|
|
|
|
seconds = now - timedelta(seconds=30)
|
|
assert pretty_date(seconds, now) == "30 seconds ago"
|
|
|
|
a_minute = now - timedelta(seconds=60)
|
|
assert pretty_date(a_minute, now) == "a minute ago"
|
|
|
|
minutes = now - timedelta(seconds=1800)
|
|
assert pretty_date(minutes, now) == "30 minutes ago"
|
|
|
|
an_hour = now - timedelta(hours=1)
|
|
assert pretty_date(an_hour, now) == "an hour ago"
|
|
|
|
hours = now - timedelta(hours=2)
|
|
assert pretty_date(hours, now) == "2 hours ago"
|
|
|
|
yesterday = now - timedelta(days=1)
|
|
assert pretty_date(yesterday, now) == "yesterday"
|
|
|
|
days = now - timedelta(days=3)
|
|
assert pretty_date(days, now) == "3 days ago"
|
|
|
|
a_week = now - timedelta(weeks=1)
|
|
assert pretty_date(a_week, now) == "a week ago"
|
|
|
|
weeks = now - timedelta(weeks=2)
|
|
assert pretty_date(weeks, now) == "2 weeks ago"
|
|
|
|
a_month = now - timedelta(days=30)
|
|
assert pretty_date(a_month, now) == "a month ago"
|
|
|
|
months = now - timedelta(days=60)
|
|
assert pretty_date(months, now) == "2 months ago"
|
|
|
|
a_year = now - timedelta(days=365)
|
|
assert pretty_date(a_year, now) == "a year ago"
|
|
|
|
years = now - timedelta(days=365 * 2)
|
|
assert pretty_date(years, now) == "2 years ago"
|
|
|
|
|
|
@pytest.mark.parametrize('delta,pretty_string', [
|
|
(timedelta(days=1), 'a day ago'),
|
|
(timedelta(days=1), 'yesterday'),
|
|
(timedelta(days=1), '1 day ago'),
|
|
(timedelta(weeks=1), '1 week ago'),
|
|
(timedelta(weeks=3), '3 weeks ago'),
|
|
(timedelta(days=30), '1 month ago'),
|
|
(timedelta(days=730), '2 years ago'),
|
|
])
|
|
def test_pretty_string_to_date_delta(now, delta, pretty_string):
|
|
t1 = now - delta
|
|
t2 = llnl.util.lang.pretty_string_to_date(pretty_string, now)
|
|
assert t1 == t2
|
|
|
|
|
|
@pytest.mark.parametrize('format,pretty_string', [
|
|
('%Y', '2018'),
|
|
('%Y-%m', '2015-03'),
|
|
('%Y-%m-%d', '2015-03-28'),
|
|
('%Y-%m-%d %H:%M', '2015-03-28 11:12'),
|
|
('%Y-%m-%d %H:%M:%S', '2015-03-28 23:34:45'),
|
|
])
|
|
def test_pretty_string_to_date(format, pretty_string):
|
|
t1 = datetime.strptime(pretty_string, format)
|
|
t2 = llnl.util.lang.pretty_string_to_date(pretty_string, now)
|
|
assert t1 == t2
|
|
|
|
|
|
def test_match_predicate():
|
|
matcher = match_predicate(lambda x: True)
|
|
assert matcher('foo')
|
|
assert matcher('bar')
|
|
assert matcher('baz')
|
|
|
|
matcher = match_predicate(['foo', 'bar'])
|
|
assert matcher('foo')
|
|
assert matcher('bar')
|
|
assert not matcher('baz')
|
|
|
|
matcher = match_predicate(r'^(foo|bar)$')
|
|
assert matcher('foo')
|
|
assert matcher('bar')
|
|
assert not matcher('baz')
|
|
|
|
with pytest.raises(ValueError):
|
|
matcher = match_predicate(object())
|
|
matcher('foo')
|
|
|
|
|
|
def test_load_modules_from_file(module_path):
|
|
# Check prerequisites
|
|
assert 'foo' not in sys.modules
|
|
|
|
# Check that the module is loaded correctly from file
|
|
foo = llnl.util.lang.load_module_from_file('foo', module_path)
|
|
assert 'foo' in sys.modules
|
|
assert foo.value == 1
|
|
assert foo.path == os.path.join('/usr', 'bin')
|
|
|
|
# Check that the module is not reloaded a second time on subsequent calls
|
|
foo.value = 2
|
|
foo = llnl.util.lang.load_module_from_file('foo', module_path)
|
|
assert 'foo' in sys.modules
|
|
assert foo.value == 2
|
|
assert foo.path == os.path.join('/usr', 'bin')
|
|
|
|
|
|
def test_uniq():
|
|
assert [1, 2, 3] == llnl.util.lang.uniq([1, 2, 3])
|
|
assert [1, 2, 3] == llnl.util.lang.uniq([1, 1, 1, 1, 2, 2, 2, 3, 3])
|
|
assert [1, 2, 1] == llnl.util.lang.uniq([1, 1, 1, 1, 2, 2, 2, 1, 1])
|
|
assert [] == llnl.util.lang.uniq([])
|