mirror of
https://github.com/gsi-upm/soil
synced 2024-11-14 15:32:29 +00:00
91 lines
2.5 KiB
Python
91 lines
2.5 KiB
Python
|
from unittest import TestCase
|
||
|
|
|
||
|
|
import os
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
from soil import history, analysis
|
||
|
|
|
||
|
|
|
||
|
|
ROOT = os.path.abspath(os.path.dirname(__file__))
|
||
|
|
|
||
|
|
|
||
|
|
class TestHistory(TestCase):
|
||
|
|
|
||
|
|
def test_history(self):
|
||
|
|
"""
|
||
|
|
"""
|
||
|
|
tuples = (
|
||
|
|
('a_0', 0, 'id', 'h', ),
|
||
|
|
('a_0', 1, 'id', 'e', ),
|
||
|
|
('a_0', 2, 'id', 'l', ),
|
||
|
|
('a_0', 3, 'id', 'l', ),
|
||
|
|
('a_0', 4, 'id', 'o', ),
|
||
|
|
('a_1', 0, 'id', 'v', ),
|
||
|
|
('a_1', 1, 'id', 'a', ),
|
||
|
|
('a_1', 2, 'id', 'l', ),
|
||
|
|
('a_1', 3, 'id', 'u', ),
|
||
|
|
('a_1', 4, 'id', 'e', ),
|
||
|
|
('env', 1, 'prob', 1),
|
||
|
|
('env', 3, 'prob', 2),
|
||
|
|
('env', 5, 'prob', 3),
|
||
|
|
('a_2', 7, 'finished', True),
|
||
|
|
)
|
||
|
|
h = history.History()
|
||
|
|
h.save_tuples(tuples)
|
||
|
|
# assert h['env', 0, 'prob'] == 0
|
||
|
|
for i in range(1, 7):
|
||
|
|
assert h['env', i, 'prob'] == ((i-1)//2)+1
|
||
|
|
|
||
|
|
|
||
|
|
for i, k in zip(range(5), 'hello'):
|
||
|
|
assert h['a_0', i, 'id'] == k
|
||
|
|
for record, value in zip(h['a_0', None, 'id'], 'hello'):
|
||
|
|
t_step, val = record
|
||
|
|
assert val == value
|
||
|
|
|
||
|
|
for i, k in zip(range(5), 'value'):
|
||
|
|
assert h['a_1', i, 'id'] == k
|
||
|
|
for i in range(5, 8):
|
||
|
|
assert h['a_1', i, 'id'] == 'e'
|
||
|
|
for i in range(7):
|
||
|
|
assert h['a_2', i, 'finished'] == False
|
||
|
|
assert h['a_2', 7, 'finished']
|
||
|
|
|
||
|
|
def test_history_gen(self):
|
||
|
|
"""
|
||
|
|
"""
|
||
|
|
tuples = (
|
||
|
|
('a_1', 0, 'id', 'v', ),
|
||
|
|
('a_1', 1, 'id', 'a', ),
|
||
|
|
('a_1', 2, 'id', 'l', ),
|
||
|
|
('a_1', 3, 'id', 'u', ),
|
||
|
|
('a_1', 4, 'id', 'e', ),
|
||
|
|
('env', 1, 'prob', 1),
|
||
|
|
('env', 2, 'prob', 2),
|
||
|
|
('env', 3, 'prob', 3),
|
||
|
|
('a_2', 7, 'finished', True),
|
||
|
|
)
|
||
|
|
h = history.History()
|
||
|
|
h.save_tuples(tuples)
|
||
|
|
for t_step, key, value in h['env', None, None]:
|
||
|
|
assert t_step == value
|
||
|
|
assert key == 'prob'
|
||
|
|
|
||
|
|
records = list(h[None, 7, None])
|
||
|
|
assert len(records) == 3
|
||
|
|
for i in records:
|
||
|
|
agent_id, key, value = i
|
||
|
|
if agent_id == 'a_1':
|
||
|
|
assert key == 'id'
|
||
|
|
assert value == 'e'
|
||
|
|
elif agent_id == 'a_2':
|
||
|
|
assert key == 'finished'
|
||
|
|
assert value == True
|
||
|
|
else:
|
||
|
|
assert key == 'prob'
|
||
|
|
assert value == 3
|
||
|
|
|
||
|
|
|
||
|
|
records = h['a_1', 7, None]
|
||
|
|
assert records['id'] == 'e'
|