Add TimeDelta and DateTime tests

C-SELLERS · C-SELLERS · commit 508db2efa934 · 2021-02-15T13:14:19.000-08:00
diff --git a/src/embed_tests/TestConverter.cs b/src/embed_tests/TestConverter.cs
@@ -58,6 +58,32 @@ public void ConvertPyListToArray()
             Assert.AreEqual(result, array);
         }
 
+        [Test]
+        public void ConvertTimeSpanRoundTrip()
+        {
+            var timespan = new TimeSpan(0, 1, 0, 0);
+            var pyTimedelta = timespan.ToPython();
+
+            object result;
+            var converted = Converter.ToManaged(pyTimedelta.Handle, typeof(TimeSpan), out result, false);
+
+            Assert.IsTrue(converted);
+            Assert.AreEqual(result, timespan);
+        }
+
+        [Test]
+        public void ConvertDateTimeRoundTrip()
+        {
+            var datetime = new DateTime(2000, 1, 1);
+            var pyDatetime = datetime.ToPython();
+
+            object result;
+            var converted = Converter.ToManaged(pyDatetime.Handle, typeof(DateTime), out result, false);
+
+            Assert.IsTrue(converted);
+            Assert.AreEqual(result, datetime);
+        }
+
         [Test]
         public void TestConvertSingleToManaged(
             [Values(float.PositiveInfinity, float.NegativeInfinity, float.MinValue, float.MaxValue, float.NaN,
diff --git a/src/testing/conversiontest.cs b/src/testing/conversiontest.cs
@@ -1,3 +1,5 @@
+using System;
+
 namespace Python.Test
 {
     using System.Collections.Generic;
@@ -31,6 +33,8 @@ public ConversionTest()
         public ShortEnum EnumField;
         public object ObjectField = null;
         public ISpam SpamField;
+        public DateTime DateTimeField;
+        public TimeSpan TimeSpanField;
 
         public byte[] ByteArrayField;
         public sbyte[] SByteArrayField;
diff --git a/tests/test_conversion.py b/tests/test_conversion.py
@@ -507,6 +507,39 @@ def test_decimal_conversion():
     with pytest.raises(TypeError):
         ConversionTest().DecimalField = 1
 
+def test_timedelta_conversion():
+    import datetime
+
+    ob = ConversionTest()
+    assert type(ob.TimeSpanField) is type(datetime.timedelta(0))
+    assert ob.TimeSpanField.days == 0
+
+    ob.TimeSpanField = datetime.timedelta(days=1)
+    assert ob.TimeSpanField.days == 1
+
+    with pytest.raises(TypeError):
+        ConversionTest().TimeSpanField = None
+
+    with pytest.raises(TypeError):
+        ConversionTest().TimeSpanField = "spam"
+
+def test_datetime_conversion():
+    from datetime import datetime
+
+    ob = ConversionTest()
+    assert type(ob.DateTimeField) is type(datetime(1,1,1))
+    assert ob.DateTimeField.day == 1
+
+    ob.DateTimeField = datetime(2000,1,2)
+    assert ob.DateTimeField.day == 2
+    assert ob.DateTimeField.month == 1
+    assert ob.DateTimeField.year == 2000
+
+    with pytest.raises(TypeError):
+        ConversionTest().DateTimeField = None
+
+    with pytest.raises(TypeError):
+        ConversionTest().DateTimeField = "spam"
 
 def test_string_conversion():
     """Test string / unicode conversion."""
@@ -716,3 +749,5 @@ def CanEncode(self, clr_type):
     l = ob.ListField
     l.Add(42)
     assert ob.ListField.Count == 1
+
+
