def test_cast_weird_strings_to_date(self):
# Mimic Spark behavior
self.assertEqual(
cast_to_date("2019-10-0001Tthis should be ignored",
StringType(),
options=BASE_OPTIONS), datetime.date(2019, 10, 1))
def test_cast_date_as_string_to_date(self):
self.assertEqual(
cast_to_date("2019-03-01", StringType(), options=BASE_OPTIONS),
datetime.date(2019, 3, 1))
def test_cast_date_without_0_as_string_to_date(self):
self.assertEqual(
cast_to_date("2019-4-1", StringType(), options=BASE_OPTIONS),
datetime.date(2019, 4, 1))
def test_cast_year_month_as_string_to_date(self):
self.assertEqual(
cast_to_date("2019-02", StringType(), options=BASE_OPTIONS),
datetime.date(2019, 2, 1))