test/test-cifar.rb in red-datasets-0.0.5 vs test/test-cifar.rb in red-datasets-0.0.6

@@ -214,6 +214,16 @@
         assert_equal([[100] * 3072, [200] * 3072],
                      @dataset.to_table[:pixels])
       end
     end
   end
+
+  sub_test_case("invalid") do
+    test("type") do
+      invalid_type = :invalid
+      message = "Please set type :train or :test: #{invalid_type.inspect}"
+      assert_raise(ArgumentError.new(message)) do
+        Datasets::CIFAR.new(type: invalid_type)
+      end
+    end
+  end
 end