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 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170
|
#!/usr/bin/env python
"""Tests for tablib._vendor.dbfpy."""
import datetime as dt
import unittest
from tablib._vendor.dbfpy import utils
class UtilsUnzfillTestCase(unittest.TestCase):
"""dbfpy.utils.unzfill test cases."""
def test_unzfill_with_nul(self):
# Arrange
text = b"abc\0xyz"
# Act
output = utils.unzfill(text)
# Assert
self.assertEqual(output, b"abc")
def test_unzfill_without_nul(self):
# Arrange
text = b"abcxyz"
# Act
output = utils.unzfill(text)
# Assert
self.assertEqual(output, b"abcxyz")
class UtilsGetDateTestCase(unittest.TestCase):
"""dbfpy.utils.getDate test cases."""
def test_getDate_none(self):
# Arrange
value = None
# Act
output = utils.getDate(value)
# Assert
self.assertIsInstance(output, dt.date)
def test_getDate_datetime_date(self):
# Arrange
value = dt.date(2019, 10, 19)
# Act
output = utils.getDate(value)
# Assert
self.assertIsInstance(output, dt.date)
self.assertEqual(output, value)
def test_getDate_datetime_datetime(self):
# Arrange
value = dt.datetime(2019, 10, 19, 12, 00, 00)
# Act
output = utils.getDate(value)
# Assert
self.assertIsInstance(output, dt.date)
self.assertEqual(output, value)
def test_getDate_datetime_timestamp(self):
# Arrange
value = 1571515306
# Act
output = utils.getDate(value)
# Assert
self.assertIsInstance(output, dt.date)
self.assertEqual(output, dt.date(2019, 10, 19))
def test_getDate_datetime_string_yyyy_mm_dd(self):
# Arrange
value = "20191019"
# Act
output = utils.getDate(value)
# Assert
self.assertIsInstance(output, dt.date)
self.assertEqual(output, dt.date(2019, 10, 19))
def test_getDate_datetime_string_yymmdd(self):
# Arrange
value = "191019"
# Act
output = utils.getDate(value)
# Assert
self.assertIsInstance(output, dt.date)
self.assertEqual(output, dt.date(2019, 10, 19))
class UtilsGetDateTimeTestCase(unittest.TestCase):
"""dbfpy.utils.getDateTime test cases."""
def test_getDateTime_none(self):
# Arrange
value = None
# Act
output = utils.getDateTime(value)
# Assert
self.assertIsInstance(output, dt.datetime)
def test_getDateTime_datetime_datetime(self):
# Arrange
value = dt.datetime(2019, 10, 19, 12, 00, 00)
# Act
output = utils.getDateTime(value)
# Assert
self.assertIsInstance(output, dt.date)
self.assertEqual(output, value)
def test_getDateTime_datetime_date(self):
# Arrange
value = dt.date(2019, 10, 19)
# Act
output = utils.getDateTime(value)
# Assert
self.assertIsInstance(output, dt.date)
self.assertEqual(output, dt.datetime(2019, 10, 19, 00, 00))
def test_getDateTime_datetime_timestamp(self):
# Arrange
value = 1571515306
# Act
output = utils.getDateTime(value)
# Assert
self.assertIsInstance(output, dt.datetime)
def test_getDateTime_datetime_string(self):
# Arrange
value = "20191019"
# Act / Assert
with self.assertRaises(NotImplementedError):
utils.getDateTime(value)
class InvalidValueTestCase(unittest.TestCase):
"""dbfpy.utils._InvalidValue test cases."""
def test_sanity(self):
# Arrange
INVALID_VALUE = utils.INVALID_VALUE
# Act / Assert
self.assertEqual(INVALID_VALUE, INVALID_VALUE)
self.assertNotEqual(INVALID_VALUE, 123)
self.assertEqual(int(INVALID_VALUE), 0)
self.assertEqual(float(INVALID_VALUE), 0.0)
self.assertEqual(str(INVALID_VALUE), "")
self.assertEqual(repr(INVALID_VALUE), "<INVALID>")
|