def test_date_field_from_dict():
str_date = "2015-01-15 00:01:59"
date = datetime.strptime(str_date, "%Y-%m-%d %H:%M:%S")
field = DateField(date_format="%Y-%m-%d %H:%M:%S")
assert field.from_dict(date) == date
assert field.from_dict(str_date) == date
with pytest.raises(ValueError) as ex:
field.from_dict(10)
assert str(ex.value) == "Expected str or date. <type 'int'> found"
def test_date_multi_field_from_dict():
str_date = "2015-01-15 00:01:59"
date = datetime.strptime(str_date, "%Y-%m-%d %H:%M:%S")
dates = [str_date, date]
field = DateField(multi=True, date_format="%Y-%m-%d %H:%M:%S")
assert field.from_dict(dates) == [date, date]
with pytest.raises(ValueError) as ex:
field.from_dict([10])
assert str(ex.value) == "Expected str or date. <type 'int'> found"
def test_date_multi_field_from_dict():
str_date = "2015-01-15 00:01:59"
date = datetime.strptime(str_date, "%Y-%m-%d %H:%M:%S")
dates = [str_date, date]
field = DateField(multi=True, date_format="%Y-%m-%d %H:%M:%S")
assert field.from_dict(dates) == [date, date]
with pytest.raises(ValueError) as ex:
field.from_dict([10])
assert str(ex.value) == "Expected str or date. " + str(int) + " found"
class DocWithArrays(Document):
_index = 'text_indice'
_doctype = 'DocWithArrays'
date_array = ArrayField(DateField())
long_array = ArrayField(LongField())
str_array = ArrayField(StringField())
empyt_array = ArrayField(StringField())
def test_date_field_from_dict():
str_date = "2015-01-15 00:01:59"
date = datetime.strptime(str_date, "%Y-%m-%d %H:%M:%S")
field = DateField(date_format="%Y-%m-%d %H:%M:%S")
assert field.from_dict(date) == date
assert field.from_dict(str_date) == date
with pytest.raises(ValueError) as ex:
field.from_dict(10)
assert str(ex.value) == "Expected str or date. " + str(int) + " found"
def test_date_field_to_dict():
date = datetime.strptime("2015-01-15 00:01:59", "%Y-%m-%d %H:%M:%S")
field = DateField(date_format="%Y-%m-%d %H:%M:%S")
assert field.to_dict(date) == "2015-01-15 00:01:59"
def test_date_field_from_dict_accept_none():
field = DateField(multi=True)
serialized = [None]
assert field.from_dict(serialized) == []
def test_date_field_to_dict():
date = datetime.strptime("2015-01-15 00:01:59", "%Y-%m-%d %H:%M:%S")
field = DateField(date_format="%Y-%m-%d %H:%M:%S")
assert field.to_dict(date) == "2015-01-15 00:01:59"
