basic_routines.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	# Copyright (C) 2018 by the RROMPy authors
	#
	# This file is part of RROMPy.
	#
	# RROMPy is free software: you can redistribute it and/or modify
	# it under the terms of the GNU Lesser General Public License as published by
	# the Free Software Foundation, either version 3 of the License, or
	# (at your option) any later version.
	#
	# RROMPy is distributed in the hope that it will be useful,
	# but WITHOUT ANY WARRANTY; without even the implied warranty of
	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	# GNU Lesser General Public License for more details.
	#
	# You should have received a copy of the GNU Lesser General Public License
	# along with RROMPy. If not, see <http://www.gnu.org/licenses/>.
	#

	import numpy as np
	from rrompy.utilities.base import (findDictStrKey, purgeDict, purgeList,
	lowDiscrepancy)

	def test_dict_list():
	dictBase = {str(x): str(x**2) for x in range(10)}
	assert findDictStrKey("-1", dictBase.keys()) is None
	assert findDictStrKey("5", dictBase.keys()) == "5"

	dictPurged = purgeDict(dictBase, [str(x) for x in range(4)], silent = True)
	assert len(dictPurged.keys()) == 4

	listBase = list(dictBase.values())
	listPurged = purgeList(listBase, [str(x**2) for x in range(4,6)],
	silent = True)
	assert len(listPurged) == 2

	def test_vandercorput():
	idxs = lowDiscrepancy(8)
	assert np.all(idxs == np.array([0,4,2,6,1,5,3,7]))