Browse Source

Unit test for Dataset shuffle

tags/v0.40-tf2.4-tstring
Sebastian Hantsch 4 years ago
parent
commit
3db1b2886f
1 changed files with 20 additions and 0 deletions
  1. +20
    -0
      test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs

+ 20
- 0
test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs View File

@@ -151,5 +151,25 @@ namespace TensorFlowNET.UnitTest.Dataset
var cardinality = dataset.dataset_cardinality(); var cardinality = dataset.dataset_cardinality();
Assert.AreEqual(new long[] { 10 }, cardinality.numpy()); Assert.AreEqual(new long[] { 10 }, cardinality.numpy());
} }

[TestMethod]
public void Shuffle()
{
tf.set_random_seed(1234);

var dataset = tf.data.Dataset.range(3);
var shuffled = dataset.shuffle(3);

var zipped = tf.data.Dataset.zip(dataset, shuffled);

bool allEqual = true;
foreach (var item in zipped)
{
if (item.Item1 != item.Item2)
allEqual = false;
}

Assert.IsFalse(allEqual);
}
} }
} }

Loading…
Cancel
Save