add test for ListDevices. (migueldeicaza#244)

alexpantyukhin · migueldeicaza · commit 206a071ed1da · 2018-04-01T22:26:32.000-04:00
* add test for ListDevices.

* free device list.
diff --git a/TensorFlowSharp/Tensorflow.cs b/TensorFlowSharp/Tensorflow.cs
@@ -2512,8 +2512,7 @@ public IEnumerable<DeviceAttributes> ListDevices(TFStatus status = null)
 				list.Add (new DeviceAttributes (name, deviceType, memory));
 			}
 
-			// TODO: Fix deleting.
-			// TF_DeleteDeviceList (rawDeviceList);
+			TF_DeleteDeviceList (rawDeviceList);
 
 			return list;
 		}
diff --git a/tests/TensorFlowSharp.Tests.CSharp/SessionTests.cs b/tests/TensorFlowSharp.Tests.CSharp/SessionTests.cs
@@ -0,0 +1,20 @@
+&#65279;using System.Linq;
+using TensorFlow;
+using Xunit;
+
+namespace TensorFlowSharp.Tests.CSharp
+{
+	public class SessionTests
+	{
+		[Fact]
+		public void Should_ListDeviceReturnDevices ()
+		{
+			using (var graph = new TFGraph ())
+			using (var session = new TFSession (graph)) {
+				var devices = session.ListDevices ();
+
+				Assert.True(devices.Any());
+			}
+		}
+	}
+}
diff --git a/tests/TensorFlowSharp.Tests.CSharp/TensorFlowSharp.Tests.CSharp.csproj b/tests/TensorFlowSharp.Tests.CSharp/TensorFlowSharp.Tests.CSharp.csproj
@@ -67,6 +67,7 @@
     <Compile Include="CondTests.cs" />
     <Compile Include="GradientTests.cs" />
     <Compile Include="ArrayTests.cs" />
+    <Compile Include="SessionTests.cs" />
     <Compile Include="ShapeTests.cs" />
     <Compile Include="TensorTests.cs" />
     <Compile Include="ClipTests.cs" />

Original file line number	Diff line number	Diff line change
`@@ -2512,8 +2512,7 @@ public IEnumerable<DeviceAttributes> ListDevices(TFStatus status = null)`
`2512`	`2512`	`list.Add (new DeviceAttributes (name, deviceType, memory));`
`2513`	`2513`	`}`
`2514`	`2514`
`2515`		`- // TODO: Fix deleting.`
`2516`		`- // TF_DeleteDeviceList (rawDeviceList);`
	`2515`	`+ TF_DeleteDeviceList (rawDeviceList);`
`2517`	`2516`
`2518`	`2517`	`return list;`
`2519`	`2518`	`}`