def test_spliter_more(self):
self.perform(
[
straw.StrawPair(
straw.spliter(lambda x: x, lambda x: x, [
straw.field("aa", straw.string()),
straw.field("ab", straw.string()),
straw.field("ac", straw.string()),
]),
straw.spliter(lambda x: x, lambda x: x, [
straw.field("ba", straw.number()),
straw.field("bb", straw.number()),
straw.field("bc", straw.number()),
]))
],
{
'aa': '1.0',
'ab': '2.0',
'ac': '3.0'
},
{
'ba': 1.,
'bb': 2.,
'bc': 3.
},
)
def test_readme(self):
self.perform(
[
straw.StrawPair(
straw.field("Name", straw.field("First", straw.string())),
straw.field("first_name", straw.string())),
straw.StrawPair(
straw.field("Name", straw.field("Last", straw.string())),
straw.field("last_name", straw.string())),
straw.StrawPair(
straw.field(
"Birthday",
straw.spliter(
lambda xs: '/'.join([str(x) for x in xs]),
lambda s: s.split('/'), [
straw.field("Year", straw.number()),
straw.field("Month", straw.number()),
straw.field("Day", straw.number()),
])), straw.field("birthday", straw.string())),
straw.StrawPair(
straw.field(
"WorkAddress",
straw.spliter(
lambda x: ', '.join(x), lambda x: x.split(', '), [
straw.field("Street", straw.string()),
straw.field("City", straw.string()),
straw.field("State", straw.string()),
straw.field("PostalCode", straw.string()),
straw.field("Country", straw.string()),
])),
straw.field('addresses', straw.field(
'work', straw.string())),
)
], {
"Name": {
"First": "Walter",
"Last": "Bishop"
},
"Birthday": {
"Year": 1946,
"Month": 8,
"Day": 20
},
"WorkAddress": {
"Street": "350 W Georgia St",
"City": "Vancouver",
"State": "BC",
"Country": "Canada",
"PostalCode": "V6B 6B1"
}
}, {
"first_name": "Walter",
"last_name": "Bishop",
"birthday": "1946/8/20",
"addresses": {
"work": "350 W Georgia St, Vancouver, BC, V6B 6B1, Canada"
}
})
def test_index(self):
self.perform(
[
straw.StrawPair(straw.index(4, straw.number()),
straw.field("a", straw.number()))
],
[None, None, None, None, 42.],
{'a': 42.},
)
def test_basic(self):
self.perform([
straw.StrawPair(straw.field("a", straw.number()),
straw.field("upstream_a", straw.number())),
straw.StrawPair(straw.field("b", straw.number()),
straw.field("upstream_b", straw.number())),
], {
"a": 1,
"b": 2,
}, {
"upstream_a": 1,
"upstream_b": 2,
})
def test_at(self):
self.perform(
[
straw.StrawPair(straw.at(["a", "b", "c"], straw.number()),
straw.at(["ab", "c"], straw.number())),
],
{'a': {
'b': {
'c': 42
}
}},
{'ab': {
'c': 42
}},
)
def test_nested_fields(self):
self.perform(
[
straw.StrawPair(
straw.field(
"a", straw.field("b", straw.field("c",
straw.number()))),
straw.field("abc", straw.number())),
],
{'a': {
'b': {
'c': 42
}
}},
{'abc': 42},
)
def test_type(self):
self.perform(
[
straw.StrawPair(straw.field("a", straw.number()),
straw.field("b", straw.string()))
],
{'a': 3.14},
{'b': "3.14"},
)