def value_with_dtype(dtype, value): """ Make a value with the specified numpy dtype. """ name = dtype.name if name.startswith('datetime64'): if name != 'datetime64[ns]': raise TypeError("Expected datetime64[ns], but got %s." % name) return make_datetime64ns(value) return dtype.type(value)
def test_make_datetime_adjustment(self): overwrite_dt = make_datetime64ns(0) result = adj.make_adjustment_from_indices( 1, 2, 3, 4, adjustment_kind=adj.OVERWRITE, value=overwrite_dt, ) expected = adj.Datetime64Overwrite( first_row=1, last_row=2, first_col=3, last_col=4, value=overwrite_dt, ) self.assertEqual(result, expected)