def test_dict_to_numpy_array1(self): d = {"a": 1, "b": 2} a = dict_to_numpy_array1(d, mapping={"a": 0, "b": 1}) np.testing.assert_allclose(a, np.array([1, 2])) a = dict_to_numpy_array1(d, mapping={"b": 0, "a": 1}) np.testing.assert_allclose(a, np.array([2, 1])) a = dict_to_numpy_array1(d) np.testing.assert_allclose(a.sum(), 3)
def test_dict_to_numpy_array1(self): d = {'a': 1, 'b': 2} a = dict_to_numpy_array1(d, mapping={'a': 0, 'b': 1}) assert_allclose(a, numpy.array([1, 2])) a = dict_to_numpy_array1(d, mapping={'b': 0, 'a': 1}) assert_allclose(a, numpy.array([2, 1])) a = dict_to_numpy_array1(d) assert_allclose(a.sum(), 3)
def test_dict_to_numpy_array_b(self): d = {"a": 1, "b": 2} mapping = {"a": 0, "b": 1} a = dict_to_numpy_array(d, mapping=mapping) assert_allclose(a, numpy.array([1, 2])) a = dict_to_numpy_array1(d) assert_allclose(a.sum(), 3)