1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
from __future__ import annotations
from typing import Any
from itemloaders import ItemLoader
from itemloaders.processors import Compose, Identity, TakeFirst
class TestOutputProcessorDict:
def test_output_processor(self):
class TempDict(dict[str, Any]):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.setdefault("temp", 0.3)
class TempLoader(ItemLoader):
default_item_class = TempDict
default_input_processor = Identity()
default_output_processor = Compose(TakeFirst())
loader = TempLoader()
item = loader.load_item()
assert isinstance(item, TempDict)
assert dict(item) == {"temp": 0.3}
class TestOutputProcessorItem:
def test_output_processor(self):
class TempLoader(ItemLoader):
default_input_processor = Identity()
default_output_processor = Compose(TakeFirst())
item: dict[str, Any] = {}
item.setdefault("temp", 0.3)
loader = TempLoader(item=item)
item = loader.load_item()
assert isinstance(item, dict)
assert dict(item) == {"temp": 0.3}
|
