def test_out_of_bounds_random_edge(self): map = WorldMap.generate_empty_map(3, 4, {}) with self.assertRaisesRegexp(ValueError, 'Beyond range'): get_random_edge_index(map, rng=ConstantRng(-1)) with self.assertRaisesRegexp(ValueError, 'Beyond range'): get_random_edge_index(map, rng=ConstantRng(6))
def test_get_random_edge_index(self): height, width = 3, 4 # First row self.assertEqual( (1, 0), get_random_edge_index(height, width, rng=ConstantRng(0))) self.assertEqual( (2, 0), get_random_edge_index(height, width, rng=ConstantRng(1))) # Last row self.assertEqual( (1, 2), get_random_edge_index(height, width, rng=ConstantRng(2))) self.assertEqual( (2, 2), get_random_edge_index(height, width, rng=ConstantRng(3))) # First column self.assertEqual( (0, 1), get_random_edge_index(height, width, rng=ConstantRng(4))) # Last column self.assertEqual( (3, 1), get_random_edge_index(height, width, rng=ConstantRng(5))) # Verify no out of bounds with self.assertRaisesRegexp(ValueError, 'Beyond range'): get_random_edge_index(height, width, rng=ConstantRng(-1)) with self.assertRaisesRegexp(ValueError, 'Beyond range'): get_random_edge_index(height, width, rng=ConstantRng(6))
def test_get_random_edge_index_can_give_all_possible(self): map = WorldMap.generate_empty_map(3, 4, {}) get_random_edge_index(map, rng=ConstantRng(1)) expected = frozenset( ((0, 1), (1, 1), (-1, 0), (2, 0), (0, -1), (1, -1))) actual = frozenset( get_random_edge_index(map, rng=ConstantRng(i)) for i in range(6)) self.assertEqual(expected, actual)
def test_get_random_edge_index_can_give_all_possible(self): map = WorldMap.generate_empty_map(3, 4, {}) get_random_edge_index(map, rng=ConstantRng(1)) expected = frozenset(( (0, 1), (1, 1), (-1, 0), (2, 0), (0, -1), (1, -1), )) actual = frozenset(get_random_edge_index(map, rng=ConstantRng(i)) for i in xrange(6)) self.assertEqual(expected, actual)
def test_get_random_edge_index(self): map = WorldMap.generate_empty_map(3, 4, {}) self.assertEqual( (0, -1), get_random_edge_index(map, rng=ConstantRng(0))) self.assertEqual( (1, -1), get_random_edge_index(map, rng=ConstantRng(1))) self.assertEqual( (0, 1), get_random_edge_index(map, rng=ConstantRng(2))) self.assertEqual( (1, 1), get_random_edge_index(map, rng=ConstantRng(3))) self.assertEqual( (-1, 0), get_random_edge_index(map, rng=ConstantRng(4))) self.assertEqual( (2, 0), get_random_edge_index(map, rng=ConstantRng(5))) # Verify no out of bounds with self.assertRaisesRegexp(ValueError, 'Beyond range'): get_random_edge_index(map, rng=ConstantRng(-1)) with self.assertRaisesRegexp(ValueError, 'Beyond range'): get_random_edge_index(map, rng=ConstantRng(6))