def test_show_with_date_query(self):
query = {'submit_date': {'$lt': mktime(now().timetuple())}}
self.__test_get_with_query_or_select(query=json.dumps(query),
num_results=self.NUM_ROWS)
query = {'submit_date': {'$gt': mktime(now().timetuple())}}
self.__test_get_with_query_or_select(query=json.dumps(query),
num_results=0)
date = mktime(datetime(2012, 2, 1, 0).timetuple())
query = {'submit_date': {'$gt': date}}
self.__test_get_with_query_or_select(query=json.dumps(query),
num_results=4)
def _test_calculator(self):
self.dframe = self.dataset.dframe()
columns = self.dframe.columns.tolist()
self.start_num_cols = len(columns)
self.added_num_cols = 0
column_labels_to_slugs = {
column_attrs[Dataset.LABEL]: (column_name) for
(column_name, column_attrs) in self.dataset.schema.items()
}
self.label_list, self.slugified_key_list = [
list(ary) for ary in zip(*column_labels_to_slugs.items())
]
for idx, formula in enumerate(self.calculations):
name = 'test-%s' % idx
Parser.validate_formula(formula, self.dataset)
calculation = Calculation()
calculation.save(self.dataset, formula, name, self.group)
self.now = now()
calculate_columns(self.dataset, [calculation])
self.column_labels_to_slugs = self.dataset.schema.labels_to_slugs
self._test_calculation_results(name, formula)
def _test_calculator(self):
self.dframe = self.dataset.dframe()
columns = self.dframe.columns.tolist()
self.start_num_cols = len(columns)
self.added_num_cols = 0
column_labels_to_slugs = {
column_attrs[Dataset.LABEL]: (column_name)
for (column_name, column_attrs) in self.dataset.schema.items()
}
self.label_list, self.slugified_key_list = [
list(ary) for ary in zip(*column_labels_to_slugs.items())
]
for idx, formula in enumerate(self.calculations):
name = 'test-%s' % idx
Parser.validate_formula(formula, self.dataset)
calculation = Calculation()
calculation.save(self.dataset, formula, name, self.group)
self.now = now()
calculate_columns(self.dataset, [calculation])
self.column_labels_to_slugs = self.dataset.schema.labels_to_slugs
self._test_calculation_results(name, formula)
def test_show_with_date_query(self):
query = {
'submit_date': {'$lt': mktime(now().timetuple())}
}
self.__test_get_with_query_or_select(
query=json.dumps(query),
num_results=self.NUM_ROWS)
query = {
'submit_date': {'$gt': mktime(now().timetuple())}
}
self.__test_get_with_query_or_select(
query=json.dumps(query),
num_results=0)
date = mktime(datetime(2012, 2, 1, 0).timetuple())
query = {
'submit_date': {'$gt': date}
}
self.__test_get_with_query_or_select(
query=json.dumps(query),
num_results=4)
def __soft_delete(cls, query):
cls.collection.update(query,
{'$set': {cls.DELETED_AT: now().isoformat()}})
def eval(self, row, dataset):
return parse_date_to_unix_time(now())
def eval(self, row, dataset):
return parse_date_to_unix_time(now())
def __soft_delete(cls, query):
cls.collection.update(query,
{'$set': {
cls.DELETED_AT: now().isoformat()
}})
