snowflake-connector-python/test/test_unit_datetime.py at master · baxen/snowflake-connector-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright (c) 2012-2018 Snowflake Computing Inc. All right reserved.
#
import time
from datetime import datetime

from snowflake.connector import sfdatetime


def test_basic_datetime_format():
    """
    Datetime format basic
    """
    # date
    value = datetime(2014, 11, 30)
    formatter = sfdatetime.SnowflakeDateTimeFormat(u'YYYY-MM-DD')
    assert formatter.format(value) == u'2014-11-30'

    # date time => date
    value = datetime(2014, 11, 30, 12, 31, 45)
    formatter = sfdatetime.SnowflakeDateTimeFormat(u'YYYY-MM-DD')
    assert formatter.format(value) == u'2014-11-30'

    # date time => date time
    value = datetime(2014, 11, 30, 12, 31, 45)
    formatter = sfdatetime.SnowflakeDateTimeFormat(
        u'YYYY-MM-DD"T"HH24:MI:SS')
    assert formatter.format(value) == u'2014-11-30T12:31:45'

    # date time => date time in microseconds with 4 precision
    value = datetime(2014, 11, 30, 12, 31, 45, microsecond=987654)
    formatter = sfdatetime.SnowflakeDateTimeFormat(
        u'YYYY-MM-DD"T"HH24:MI:SS.FF4')
    assert formatter.format(value) == u'2014-11-30T12:31:45.9876'

    # date time => date time in microseconds with full precision up to
    # microseconds
    value = datetime(2014, 11, 30, 12, 31, 45, microsecond=987654)
    formatter = sfdatetime.SnowflakeDateTimeFormat(
        u'YYYY-MM-DD"T"HH24:MI:SS.FF')
    assert formatter.format(value) == u'2014-11-30T12:31:45.987654'


def test_datetime_with_smaller_milliseconds():
    # date time => date time in microseconds with full precision up to
    # microseconds
    value = datetime(2014, 11, 30, 12, 31, 45, microsecond=123)
    formatter = sfdatetime.SnowflakeDateTimeFormat(
        u'YYYY-MM-DD"T"HH24:MI:SS.FF9')
    assert formatter.format(value) == u'2014-11-30T12:31:45.000123'


def test_datetime_format_negative():
    u"""Datetime format negative"""
    value = datetime(2014, 11, 30, 12, 31, 45, microsecond=987654)
    formatter = sfdatetime.SnowflakeDateTimeFormat(
        u'YYYYYYMMMDDDDD"haha"hoho"hihi"H12HHH24MI')
    assert formatter.format(value) == u'20141411M3030DhahaHOHOhihiH1212H2431'


def test_struct_time_format():
    # struct_time for general use
    value = time.strptime("30 Sep 01 11:20:30", "%d %b %y %H:%M:%S")
    formatter = sfdatetime.SnowflakeDateTimeFormat(
        u'YYYY-MM-DD"T"HH24:MI:SS.FF')
    assert formatter.format(value) == '2001-09-30T11:20:30.0'

    # struct_time encapsulated in SnowflakeDateTime. Mainly used by SnowSQL
    value = sfdatetime.SnowflakeDateTime(
        time.strptime("30 Sep 01 11:20:30", "%d %b %y %H:%M:%S"),
        nanosecond=0, scale=1
    )
    formatter = sfdatetime.SnowflakeDateTimeFormat(
        u'YYYY-MM-DD"T"HH24:MI:SS.FF',
        datetime_class=sfdatetime.SnowflakeDateTime)
    assert formatter.format(value) == '2001-09-30T11:20:30.0'

    # format without fraction of seconds
    formatter = sfdatetime.SnowflakeDateTimeFormat(
        u'YYYY-MM-DD"T"HH24:MI:SS',
        datetime_class=sfdatetime.SnowflakeDateTime)
    assert formatter.format(value) == '2001-09-30T11:20:30'

    # extreme large epoch time
    value = sfdatetime.SnowflakeDateTime(
        time.gmtime(14567890123567), nanosecond=0, scale=1)
    formatter = sfdatetime.SnowflakeDateTimeFormat(
        u'YYYY-MM-DD"T"HH24:MI:SS.FF',
        datetime_class=sfdatetime.SnowflakeDateTime)
    assert formatter.format(value) == '463608-01-23T09:26:07.0'