You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

StringsApiTest.cs 1.3 kB

5 years ago
12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. using Microsoft.VisualStudio.TestTools.UnitTesting;
  2. using System;
  3. using System.Collections.Generic;
  4. using System.Text;
  5. using static Tensorflow.Binding;
  6. namespace Tensorflow.UnitTest.TF_API
  7. {
  8. [TestClass]
  9. public class StringsApiTest
  10. {
  11. [TestMethod]
  12. public void StringEqual()
  13. {
  14. var str1 = tf.constant("Hello1");
  15. var str2 = tf.constant("Hello2");
  16. var result = tf.equal(str1, str2);
  17. Assert.IsFalse(result.ToScalar<bool>());
  18. var str3 = tf.constant("Hello1");
  19. result = tf.equal(str1, str3);
  20. Assert.IsTrue(result.ToScalar<bool>());
  21. var str4 = tf.strings.substr(str1, 0, 5);
  22. var str5 = tf.strings.substr(str2, 0, 5);
  23. result = tf.equal(str4, str5);
  24. Assert.IsTrue(result.ToScalar<bool>());
  25. }
  26. [TestMethod]
  27. public void ImageType()
  28. {
  29. var imgPath = TestHelper.GetFullPathFromDataDir("shasta-daisy.jpg");
  30. var contents = tf.io.read_file(imgPath);
  31. var substr = tf.strings.substr(contents, 0, 3);
  32. var jpg = Encoding.UTF8.GetString(new byte[] { 0xff, 0xd8, 0xff });
  33. var jpg_tensor = tf.constant(jpg);
  34. var result = math_ops.equal(substr, jpg_tensor);
  35. }
  36. }
  37. }