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))