def test_brick_next_state_both_falling(self):
brick = Brick(1, 2, 0)
brick.next_state()
self.assertEqual(brick.state, 1)
def test_brick_next_state_left_falling(self):
brick = Brick(1, 2, 1)
brick.next_state()
self.assertEqual(brick.state, 2)
def test_brick_next_state_right_falling(self):
brick = Brick(1, 1, 0)
brick.next_state()
self.assertEqual(brick.state, 0)
def test_brick_next_state_none_falling(self):
brick = Brick(1, 1, 1)
brick.next_state()
self.assertEqual(brick.state, 1)