(self)
| 232 | self.assertTupleEqual(result.shape, (1, 101, 102, 103)) |
| 233 | |
| 234 | def test_one_of(self): |
| 235 | p = OneOf((A(), B(), C()), (1, 2, 1)) |
| 236 | counts = [0] * 3 |
| 237 | for _i in range(10000): |
| 238 | out = p(1.0) |
| 239 | counts[int(out - 2)] += 1 |
| 240 | self.assertAlmostEqual(counts[0] / 10000, 0.25, delta=1.0) |
| 241 | self.assertAlmostEqual(counts[1] / 10000, 0.50, delta=1.0) |
| 242 | self.assertAlmostEqual(counts[2] / 10000, 0.25, delta=1.0) |
| 243 | |
| 244 | |
| 245 | TEST_ONEOF_EXTENDED_TEST_CASES = [ |