(self)
| 353 | self.assertEqual(x.shape, (1, 1)) |
| 354 | |
| 355 | def test_complex_normal(self): |
| 356 | sample = mx.random.normal(tuple(), dtype=mx.complex64) |
| 357 | self.assertEqual(sample.shape, tuple()) |
| 358 | self.assertEqual(sample.dtype, mx.complex64) |
| 359 | |
| 360 | sample = mx.random.normal((1, 2, 3, 4), dtype=mx.complex64) |
| 361 | self.assertEqual(sample.shape, (1, 2, 3, 4)) |
| 362 | self.assertEqual(sample.dtype, mx.complex64) |
| 363 | |
| 364 | sample = mx.random.normal((1, 2, 3, 4), dtype=mx.complex64, scale=2.0, loc=3.0) |
| 365 | self.assertEqual(sample.shape, (1, 2, 3, 4)) |
| 366 | self.assertEqual(sample.dtype, mx.complex64) |
| 367 | |
| 368 | sample = mx.random.normal( |
| 369 | (1, 2, 3, 4), dtype=mx.complex64, scale=2.0, loc=3.0 + 1j |
| 370 | ) |
| 371 | self.assertEqual(sample.shape, (1, 2, 3, 4)) |
| 372 | self.assertEqual(sample.dtype, mx.complex64) |
| 373 | |
| 374 | def test_broadcastable_scale_loc(self): |
| 375 | b = mx.random.normal((10, 2)) |
nothing calls this directly
no outgoing calls
no test coverage detected