def test_date(self):
enc = QATrackJSONEncoder()
dt = datetime.date(2020, 2, 29)
assert enc.default(dt) == "29 Feb 2020"
def test_pd_df(self):
enc = QATrackJSONEncoder()
d = {'col1': [1, 2], 'col2': [3, 4]}
df = pd.DataFrame(data=d)
expected = {'col1': {0: 1, 1: 2}, 'col2': {0: 3, 1: 4}}
assert enc.default(df) == expected
def test_datetime(self):
enc = QATrackJSONEncoder()
tz = pytz.timezone("America/Toronto")
dt = timezone.datetime(2020, 2, 29, 12, 34, 56, tzinfo=tz)
assert enc.default(dt) == "29 Feb 2020 12:34:56"
def test_set(self):
enc = QATrackJSONEncoder()
assert set(enc.default(set(range(3)))) == set(range(3))
def test_zip(self):
enc = QATrackJSONEncoder()
assert enc.default(zip(range(3), range(3))) == [(0, 0), (1, 1), (2, 2)]
def test_range(self):
enc = QATrackJSONEncoder()
assert enc.default(range(3)) == [0, 1, 2]
def test_np_array(self):
enc = QATrackJSONEncoder()
assert enc.default(np.array(range(3))) == [0, 1, 2]
def test_np_int(self):
enc = QATrackJSONEncoder()
assert enc.default(np.int8(1)) == 1