(self)
| 1609 | assert isinstance(v.dtype, pd.CategoricalDtype) |
| 1610 | |
| 1611 | def test_squeeze(self): |
| 1612 | v = Variable(["x", "y"], [[1]]) |
| 1613 | assert_identical(Variable([], 1), v.squeeze()) |
| 1614 | assert_identical(Variable(["y"], [1]), v.squeeze("x")) |
| 1615 | assert_identical(Variable(["y"], [1]), v.squeeze(["x"])) |
| 1616 | assert_identical(Variable(["x"], [1]), v.squeeze("y")) |
| 1617 | assert_identical(Variable([], 1), v.squeeze(["x", "y"])) |
| 1618 | |
| 1619 | v = Variable(["x", "y"], [[1, 2]]) |
| 1620 | assert_identical(Variable(["y"], [1, 2]), v.squeeze()) |
| 1621 | assert_identical(Variable(["y"], [1, 2]), v.squeeze("x")) |
| 1622 | with pytest.raises(ValueError, match=r"cannot select a dimension"): |
| 1623 | v.squeeze("y") |
| 1624 | |
| 1625 | def test_get_axis_num(self) -> None: |
| 1626 | v = Variable(["x", "y", "z"], np.random.randn(2, 3, 4)) |
nothing calls this directly
no test coverage detected