def test_input_min_date():
with testing.postgresql.Postgresql() as psql:
engine = sqlalchemy.create_engine(psql.url())
engine.execute("create table events (entity_id int, date date, outcome bool)")
for event in events_data:
engine.execute("insert into events values (%s, %s, %s::bool)", event)
engine.execute("create table states (entity_id int, date date)")
for state in state_data:
engine.execute("insert into states values (%s, %s)", state)
agg = Aggregate(
"outcome::int",
["sum", "avg"],
{
"coltype": "aggregate",
"avg": {"type": "mean"},
"sum": {"type": "constant", "value": 3},
"max": {"type": "zero"},
},
)
st = SpacetimeAggregation(
aggregates=[agg],
from_obj="events",
groups=["entity_id"],
intervals=["all"],
dates=["2016-01-01"],
state_table="states",
state_group="entity_id",
date_column='"date"',
input_min_date="2015-11-10",
)
st.execute(engine.connect())
r = engine.execute("select * from events_entity_id order by entity_id")
rows = [x for x in r]
assert rows[0]["entity_id"] == 1
assert rows[0]["date"] == date(2016, 1, 1)
assert rows[0]["events_entity_id_all_outcome::int_sum"] == 1
assert rows[0]["events_entity_id_all_outcome::int_avg"] == 1
assert rows[1]["entity_id"] == 4
assert rows[1]["date"] == date(2016, 1, 1)
assert rows[1]["events_entity_id_all_outcome::int_sum"] == 0
assert rows[1]["events_entity_id_all_outcome::int_avg"] == 0
assert len(rows) == 2
st = SpacetimeAggregation(
aggregates=[agg],
from_obj="events",
groups=["entity_id"],
intervals=["1y", "all"],
dates=["2016-01-01", "2015-01-01"],
state_table="states",
state_group="entity_id",
date_column='"date"',
input_min_date="2014-11-10",
)
with pytest.raises(ValueError):
st.validate(engine.connect())
with pytest.raises(ValueError):
st.execute(engine.connect())
def test_input_min_date():
with testing.postgresql.Postgresql() as psql:
engine = sqlalchemy.create_engine(psql.url())
engine.execute(
'create table events (entity_id int, date date, outcome bool)'
)
for event in events_data:
engine.execute(
'insert into events values (%s, %s, %s::bool)',
event
)
engine.execute(
'create table states (entity_id int, date date)'
)
for state in state_data:
engine.execute(
'insert into states values (%s, %s)',
state
)
agg = Aggregate('outcome::int', ['sum','avg'], {
"coltype": "aggregate",
"avg": {"type": "mean"},
"sum": {"type": "constant", "value": 3},
"max": {"type": "zero"}
})
st = SpacetimeAggregation(
aggregates = [agg],
from_obj = 'events',
groups = ['entity_id'],
intervals = ['all'],
dates = ['2016-01-01'],
state_table = 'states',
state_group = 'entity_id',
date_column = '"date"',
input_min_date = '2015-11-10'
)
st.execute(engine.connect())
r = engine.execute('select * from events_entity_id order by entity_id')
rows = [x for x in r]
assert rows[0]['entity_id'] == 1
assert rows[0]['date'] == date(2016, 1, 1)
assert rows[0]['events_entity_id_all_outcome::int_sum'] == 1
assert rows[0]['events_entity_id_all_outcome::int_avg'] == 1
assert rows[1]['entity_id'] == 4
assert rows[1]['date'] == date(2016, 1, 1)
assert rows[1]['events_entity_id_all_outcome::int_sum'] == 0
assert rows[1]['events_entity_id_all_outcome::int_avg'] == 0
assert len(rows) == 2
st = SpacetimeAggregation(
aggregates = [agg],
from_obj = 'events',
groups = ['entity_id'],
intervals = ['1y', 'all'],
dates = ['2016-01-01', '2015-01-01'],
state_table = 'states',
state_group = 'entity_id',
date_column = '"date"',
input_min_date = '2014-11-10'
)
with pytest.raises(ValueError):
st.validate(engine.connect())
with pytest.raises(ValueError):
st.execute(engine.connect())