more tests

Author: hugo

pandas/tests/indexing/test_datetime.py

@@ -8,14 +8,40 @@
 
 class TestDatetimeIndex(object):
 
+    def test_tuple_setitem_with_datetime_tz(self):
+        # 16889
+        # support setitem with datetimeindex with tz
+        mask = np.array([True, False, True, False])
+        idx = pd.date_range('20010101', periods=4, tz='UTC')
+        df = pd.DataFrame({'a': np.arange(4)}, index=idx).astype('float64')
+        result = df.copy()
+        result.loc[mask, :] = df.loc[mask, :]
+        tm.assert_frame_equal(result, df)
+
     def test_setitem_with_datetime_tz(self):
         mask = np.array([True, False, True, False])
         idx = pd.date_range('20010101', periods=4, tz='UTC')
         df = pd.DataFrame({'a': np.arange(4)}, index=idx).astype('float64')
-        test = df.copy()
-        test.loc[mask, :] = df.loc[mask, :]
-        tm.assert_frame_equal(test, df)
+        result = df.copy()
+        result.loc[mask] = df.loc[mask]
+        tm.assert_frame_equal(result, df)
 
+    def test_tuple_setitem_with_datetime(self):
+        mask = np.array([True, False, True, False])
+        idx = pd.date_range('20010101', periods=4)
+        df = pd.DataFrame({'a': np.arange(4)}, index=idx).astype('float64')
+        result = df.copy()
+        result.loc[mask] = df.loc[mask]
+        tm.assert_frame_equal(result, df)
+        
+    def test_setitem_datetime(self):
+        mask = np.array([True, False, True, False])
+        idx = pd.date_range('20010101', periods=4)
+        df = pd.DataFrame({'a': np.arange(4)}, index=idx).astype('float64')
+        result = df.copy()
+        result.loc[mask] = df.loc[mask]
+        tm.assert_frame_equal(result, df)
+        
     def test_indexing_with_datetime_tz(self):
 
         # 8260