From fe40f9886eb702cc8b49b3262a49e9da0865d763 Mon Sep 17 00:00:00 2001
From: Otto Fajardo <otto.fajardob@gmail.com>
Date: Mon, 30 Jan 2023 16:21:40 +0100
Subject: [PATCH] added unittest for extra_date_formats

---
 test_data/basic/date_test.sas7bdat | Bin 0 -> 131072 bytes
 tests/test_basic.py                |   9 ++++++++-
 2 files changed, 8 insertions(+), 1 deletion(-)
 create mode 100644 test_data/basic/date_test.sas7bdat

diff --git a/test_data/basic/date_test.sas7bdat b/test_data/basic/date_test.sas7bdat
new file mode 100644
index 0000000000000000000000000000000000000000..c51ad38cc43362fec063f85e618175b6f5951409
GIT binary patch
literal 131072
zcmeIzeP~s69LMqRb*r`}vFRY`ftEdxBzE`E+LqM0bZ;5BT6#5Sie|G6A6jvVMM0<j
z8a-O{M-L)FA^u@Dds4#kBtj1eX_lEzkTepZ{s_r7*KWUu@7X<DGAaln-oJ6~?|Xjt
zcRuHLUgw-Uky6`v;)$j2*4BLa(Tf#Q`Dy(#Q!c5lnL1S(8&eY+%15S^rACc6rY>ug
z8$_zhQt_0znYnd0-#R;&&E#`S^0~SBx_{Qp@fmxh&7HRB$Dvy`9L~hOJ6GMNpw!#b
zr_5iewa!R4rR&=p7Bn|4Y3Z1Ad+D3MSJ`tepVa$Gyp272^<Z189UHZ_v6I+G009IL
zKmY**5I_I{1Q0*~0R#|0009ILKmY**5I_I{1Q0*~0R#|0009ILKmY**5I_I{1Q0*~
z0R#|0009ILKmY**5I_I{1Q0*~0R#|0009ILKmY**5I_I{1Q0*~0R#|0009ILKmY**
z5I_I{1Q0*~0R#|0009ILKmY**5I_I{1Q0*~0R#|0009ILKmY**5I_I{1Q0*~0R#|0
z009ILKmY**5I_I{1Q0*~0R#|0009ILKmY**5I_I{1Q0*~0R#|0009ILKmY**5I_I{
z1Q0*~0R#|0009ILKmY**5I_I{1Q0*~0R#|0009ILKmY**5I_I{1Q0*~0R#|0009IL
zKmY**5I_I{1Q0*~0R#|0009ILKmY**5I_I{1Q0*~0R#|0009ILKmY**5I_I{1pea!
zs|MRz<0a;PG1Y`Cq{ei0!y-pg{+n7;Pt-}rhD6djewC@W+i|nM`?Sa+t!JA$)VkW#
zGqvs?6lt=xX&V!DAyEg3dSaqJFi<-7%|zXusArC;?Rj*mwiKE5d|<oq;{C3d^0)nd
z?{2@p!H)+&9ku&6HErzAn10@x`TJ}yQI8(i;d4gz1I>BP=LD&512_Mq%?Z}~oZy3<
znlp0#1wKc<*4)Q5=Ov#LF7i3Skj)7Pd`@`&`FHyD1UtWR=aHp8C%DGXmu;H!s?UkG
z>UzqKnfX!vT6bMh&q~da`!r{T&k3K>oRgaKy828$;LaCR*&ONfIdX7^=0sJR)9rJj
zA2er5pF1wxpgwDSPBh8p1Xay$J;C+03;cD3eP6rz(&KX^<L67W=4?`*&viW)*qrcQ
zpA)WIp*g`k%?W)@uugOGn)8nOT>hOqUv#PS`NZc$zi-zZ8QkaB)9Z7>Q=0RT=4?}+
z(B}lxoX^GA`R8XzO{YFT>osSs`poh3Wv%9XtUiZyJ!$9D<8y*&x#om7Y0k^)vsrUC
zYtAn9X+G%I9ke)~eLg3=^h3=Fj(_E^>vi?1^z&u6=6s<(FZl83M&~o5#Xmn!p106n
zmprRE8`Y;vb53Z^9`*T2*K>>WdCKR=v1OVQ)oV_l`s~o0iY;!P;ePd*b;#{Uw>h65
zd`^(tsyX5Qz5cqksn3OezFes}{p$0E9}ng`pP<!USGb|-E`7fq)tpFu9@d<!=6tU{
z!@8cu&gV6s6P>eEbEHmlK2x8)nzKxEj;qhYez)#mx$`;YbL9T_G$-1%$6wbT^_lMH
zOQ<<NtIzv>JX+;^=Ct|u>x+s-`hMN7IS16IS98{D&PnyDKJ3?1a6WJOobc+!niHI-
zIY-pzsOD_foI&+j>2sn-olnK}{`q-ki{{9^U;6738UhF)fB*srAb<b@2q1s}0tg_0
z00IagfB*srAb<b@2q1s}0tg_000IagfB*srAb<b@2q1s}0tg_000IagfB*srAb<b@
z2q1s}0tg_000IagfB*srAb<b@2q1s}0tg_000IagfB*srAb<b@2q1s}0tg_000Iag
zfB*srAb<b@2q1s}0tg_000IagfB*srAb<b@2q1s}0tg_000IagfB*srAb<b@2q1s}
z0tg_000IagfB*srAb<b@2q1s}0tg_000IagfB*srAb<b@2q1s}0tg_000IagfB*sr
zAb<b@2q1s}0tg_000IagfB*srAb<b@2q1s}0tg_000IagfB*srAb<b@2q1s}0tg_0
z00IagfB*srAb<b@2q1s}0tg`R{}LEJT`a~|RBnp1;vLs-k}}h86#Gr7C6B2w!*(mR
zby8t=UUHwM(d039w7T?_se0Y)P|g`Onmop4+O5=j+^P!Wtj(s4PF&$Ky2d%?&*xbn
zXPcZ3_gOl7a&70DVY`)D*3F(d&M}o{vfWCp$*n3q&f4r<7=0&7M<=V{VzHQ({>Sau
zRc5P{DoKr;?YCmN44ZpiDTBq+#ox``fsrRbWXS9n5@F7~JmuWt+TOdko|35R616>V
zJSUx~TM~7Us7>jbIeoDs-jU1od7Q4--J<<$?U!ob?VeMS$FyIo{hKmjBum6zYhPl&
zZcNK&@|n5$Tu1lag*6YDd0mBj>djcrZl-R?X4~64J4NE&j5W5mXR~%J?#)<}ot3$*
zGn>!mbA{FQlE1Y*XSc3Gz2sV&)AiZB==beig$A+t_WQVh<ROvHTxL#Z=N+Zhnj2%+
zZtwr-XSy`yOw~l4Z0~EzOt8;#+*=P@+swE(LvGCe4%ji<+vmyFHY)DT&qm^@F<+QA
zc!sfoGmQD!|L*6nrp52qDbt=d55i<UWlef&`7Be{CYBJ-wa;yx8@E>~@}T(vG?(C-
z8~BfXtQu@<{i{o|S7hG@+gEEpN&73bckisJ|5~fi_s`C*_3yhoaH6X(N{oz`S8i66
W8uOn=<VpLl$F<~r(xFlRrv3n#9d_OT

literal 0
HcmV?d00001

diff --git a/tests/test_basic.py b/tests/test_basic.py
index f0f0cc6..9ab1e89 100644
--- a/tests/test_basic.py
+++ b/tests/test_basic.py
@@ -15,7 +15,7 @@
 # limitations under the License.
 # #############################################################################
 
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, date
 import unittest
 import os
 import sys
@@ -909,6 +909,13 @@ def test_sas7bdat_file_label_linux(self):
         self.assertEqual(meta.file_label, "mytest label")
         self.assertEqual(meta.table_name, "TEST_DATA")
 
+    def test_sas7bdat_extra_date_formats(self):
+        "testing extra date format argument"
+        path = os.path.join(self.basic_data_folder, "date_test.sas7bdat")
+        df, meta = pyreadstat.read_sas7bdat(path, extra_date_formats=["MMYY", "YEAR"])
+        self.assertEqual(df['yr'].iloc[0], date(2023,1,1))
+        self.assertEqual(df['dtc4'].iloc[0], date(2023,7,1))
+
     def test_sas7bdat_file_label_windows(self):
         "testing file label for file produced on windows"
         path = os.path.join(self.basic_data_folder, "test_file_label_win.sas7bdat")