(self)
| 41 | class MetricsTest(test.TestCase): |
| 42 | |
| 43 | def testMean(self): |
| 44 | m = metrics.Mean() |
| 45 | m([1, 10, 100]) |
| 46 | m(1000) |
| 47 | m([10000.0, 100000.0]) |
| 48 | self.assertEqual(111111.0/6, m.result().numpy()) |
| 49 | self.assertEqual(dtypes.float64, m.dtype) |
| 50 | self.assertEqual(dtypes.float64, m.result().dtype) |
| 51 | |
| 52 | def testVariableCollections(self): |
| 53 | with context.graph_mode(), ops.Graph().as_default(): |