def test_three_dims(self): tiled = _TileColors(self.blue, 1) expected_tiled = np.array([[0, 0, 0], [0, 0, 0], [255, 255, 255]]) self.assertTrue(np.allclose(tiled, expected_tiled))
def test_one_dim(self): tiled = _TileColors(self.red, 1) expected_tiled = np.array([[255], [0], [0]]) self.assertTrue(np.allclose(tiled, expected_tiled))