(self)
| 265 | self.assertEqual(output_value, 2) |
| 266 | |
| 267 | def test_loop(self): |
| 268 | workspace.FeedBlob("cond", np.array(True)) |
| 269 | workspace.FeedBlob("ONE", np.array(1)) |
| 270 | workspace.FeedBlob("TWO", np.array(2)) |
| 271 | workspace.FeedBlob("TEN", np.array(10)) |
| 272 | workspace.FeedBlob("counter", np.array(0)) |
| 273 | workspace.FeedBlob("output_blob", np.array(0)) |
| 274 | |
| 275 | loop_model = ModelHelper(name="loop_test_model") |
| 276 | loop_model.net.Add(["output_blob", "TWO"], "output_blob") |
| 277 | |
| 278 | cond_model = ModelHelper(name="cond_test_model") |
| 279 | cond_model.net.Add(["counter", "ONE"], "counter") |
| 280 | comp_res = cond_model.net.LT(["counter", "TEN"]) |
| 281 | cond_model.net.Copy(comp_res, "cond") |
| 282 | |
| 283 | model = ModelHelper(name="test_model") |
| 284 | brew.loop( |
| 285 | model=model, |
| 286 | cond_blob="cond", |
| 287 | external_blobs=["cond", "ONE", "TWO", "TEN", "counter", "output_blob"], |
| 288 | loop_model=loop_model, |
| 289 | cond_model=cond_model) |
| 290 | |
| 291 | workspace.RunNetOnce(model.param_init_net) |
| 292 | workspace.RunNetOnce(model.net) |
| 293 | output_value = workspace.FetchBlob("output_blob") |
| 294 | self.assertEqual(output_value, 18) |
| 295 | |
| 296 | |
| 297 | @unittest.skipIf(not workspace.has_gpu_support, "No gpu support.") |
nothing calls this directly
no test coverage detected