From c7b04c697bf923d13d139a8b162c5dd7d888396c Mon Sep 17 00:00:00 2001
From: Jeff Donahue <jeff.donahue@gmail.com>
Date: Mon, 12 Jan 2015 13:42:51 -0800
Subject: [PATCH] Add unit test for LayerRegistry::CreateLayer

---
 src/caffe/test/test_layer_factory.cpp | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 src/caffe/test/test_layer_factory.cpp

diff --git a/src/caffe/test/test_layer_factory.cpp b/src/caffe/test/test_layer_factory.cpp
new file mode 100644
index 0000000..3ad635a
--- /dev/null
+++ b/src/caffe/test/test_layer_factory.cpp
@@ -0,0 +1,33 @@
+#include <map>
+#include <string>
+
+#include "gtest/gtest.h"
+
+#include "caffe/common.hpp"
+#include "caffe/layer.hpp"
+#include "caffe/layer_factory.hpp"
+
+#include "caffe/test/test_caffe_main.hpp"
+
+namespace caffe {
+
+template <typename TypeParam>
+class LayerFactoryTest : public MultiDeviceTest<TypeParam> {};
+
+TYPED_TEST_CASE(LayerFactoryTest, TestDtypesAndDevices);
+
+TYPED_TEST(LayerFactoryTest, TestCreateLayer) {
+  typedef typename TypeParam::Dtype Dtype;
+  typename LayerRegistry<Dtype>::CreatorRegistry& registry =
+      LayerRegistry<Dtype>::Registry();
+  shared_ptr<Layer<Dtype> > layer;
+  LayerParameter layer_param;
+  for (typename LayerRegistry<Dtype>::CreatorRegistry::iterator iter =
+       registry.begin(); iter != registry.end(); ++iter) {
+    layer_param.set_type(iter->first);
+    layer.reset(LayerRegistry<Dtype>::CreateLayer(layer_param));
+    EXPECT_EQ(iter->first, layer->type());
+  }
+}
+
+}  // namespace caffe
-- 
2.7.4