test/test-penn-treebank.rb in red-datasets-0.0.6 vs test/test-penn-treebank.rb in red-datasets-0.0.7

@@ -7,12 +7,12 @@
     test("train") do
       dataset = Datasets::PennTreebank.new(type: :train)
       records = dataset.to_a
       assert_equal([
                      887521,
-                     record("aer", 0),
-                     record("<unk>", 25),
+                     record("aer"),
+                     record("<unk>"),
                    ],
                    [
                      records.size,
                      records[0],
                      records[-1],
@@ -22,12 +22,12 @@
     test("test") do
       dataset = Datasets::PennTreebank.new(type: :test)
       records = dataset.to_a
       assert_equal([
                      78669,
-                     record("no", 0),
-                     record("us", 953),
+                     record("no"),
+                     record("us"),
                    ],
                    [
                      records.size,
                      records[0],
                      records[-1],
@@ -37,11 +37,11 @@
     test("valid") do
       dataset = Datasets::PennTreebank.new(type: :valid)
       records = dataset.to_a
       assert_equal([
                      70390,
-                     record("consumers", 0),
-                     record("N", 28),
+                     record("consumers"),
+                     record("N"),
                    ],
                    [
                      records.size,
                      records[0],
                      records[-1],