57 lines
1.1 KiB
Python
57 lines
1.1 KiB
Python
from pathlib import Path
|
|
import pytz
|
|
from my.core.cfg import tmp_config
|
|
import pytest
|
|
from my.body.weight import from_orgmode
|
|
|
|
|
|
def test_body_weight() -> None:
|
|
weights = [0.0 if isinstance(x, Exception) else x.value for x in from_orgmode()]
|
|
|
|
assert weights == [
|
|
0.0,
|
|
62.0,
|
|
0.0,
|
|
61.0,
|
|
62.0,
|
|
0.0,
|
|
]
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def prepare(tmp_path: Path):
|
|
ndir = tmp_path / 'notes'
|
|
ndir.mkdir()
|
|
logs = ndir / 'logs.org'
|
|
logs.write_text(
|
|
'''
|
|
#+TITLE: Stuff I'm logging
|
|
|
|
* Weight (org-capture) :weight:
|
|
** [2020-05-01 Fri 09:00] 62
|
|
** 63
|
|
this should be ignored, got no timestamp
|
|
** [2020-05-03 Sun 08:00] 61
|
|
** [2020-05-04 Mon 10:00] 62
|
|
'''
|
|
)
|
|
misc = ndir / 'misc.org'
|
|
misc.write_text(
|
|
'''
|
|
Some misc stuff
|
|
|
|
* unrelated note :weight:whatever:
|
|
'''
|
|
)
|
|
|
|
class orgmode:
|
|
paths = [ndir]
|
|
|
|
class weight:
|
|
# TODO ugh. this belongs to tz provider or global config or something
|
|
default_timezone = pytz.timezone('Europe/London')
|
|
|
|
with tmp_config() as cfg:
|
|
cfg.orgmode = orgmode
|
|
cfg.weight = weight
|
|
yield
|