mirror of https://github.com/apache/superset.git
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
from datetime import datetime, date, timedelta, time
|
|
from decimal import Decimal
|
|
from superset.utils import (
|
|
json_int_dttm_ser, json_iso_dttm_ser, base_json_conv, parse_human_timedelta
|
|
)
|
|
import unittest
|
|
import uuid
|
|
|
|
from mock import Mock, patch
|
|
import numpy
|
|
|
|
|
|
class UtilsTestCase(unittest.TestCase):
|
|
def test_json_int_dttm_ser(self):
|
|
dttm = datetime(2020, 1, 1)
|
|
ts = 1577836800000.0
|
|
assert json_int_dttm_ser(dttm) == ts
|
|
assert json_int_dttm_ser(date(2020, 1, 1)) == ts
|
|
assert json_int_dttm_ser(datetime(1970, 1, 1)) == 0
|
|
assert json_int_dttm_ser(date(1970, 1, 1)) == 0
|
|
assert json_int_dttm_ser(dttm + timedelta(milliseconds=1)) == (ts + 1)
|
|
|
|
with self.assertRaises(TypeError):
|
|
json_int_dttm_ser("this is not a date")
|
|
|
|
def test_json_iso_dttm_ser(self):
|
|
dttm = datetime(2020, 1, 1)
|
|
dt = date(2020, 1, 1)
|
|
t = time()
|
|
assert json_iso_dttm_ser(dttm) == dttm.isoformat()
|
|
assert json_iso_dttm_ser(dt) == dt.isoformat()
|
|
assert json_iso_dttm_ser(t) == t.isoformat()
|
|
|
|
with self.assertRaises(TypeError):
|
|
json_iso_dttm_ser("this is not a date")
|
|
|
|
def test_base_json_conv(self):
|
|
assert isinstance(base_json_conv(numpy.bool_(1)), bool) == True
|
|
assert isinstance(base_json_conv(numpy.int64(1)), int) == True
|
|
assert isinstance(base_json_conv(set([1])), list) == True
|
|
assert isinstance(base_json_conv(Decimal('1.0')), float) == True
|
|
assert isinstance(base_json_conv(uuid.uuid4()), str) == True
|
|
|
|
@patch('superset.utils.datetime')
|
|
def test_parse_human_timedelta(self, mock_now):
|
|
mock_now.return_value = datetime(2016, 12, 1)
|
|
self.assertEquals(parse_human_timedelta('now'), timedelta(0))
|