def test_mdr_transform():
"""Ensure that the MDR 'transform' function maps a new set of feature instances to the desired labels"""
features = np.array([[2, 0], [0, 0], [0, 1], [0, 0], [0, 0], [0,
0], [0, 1],
[0, 0], [0, 0], [0, 1], [0, 0], [0, 0], [0, 0],
[1, 1], [1, 1]])
classes = np.array([1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0])
mdr = MDR()
mdr.fit(features, classes)
test_features = np.array([[2, 2], [1, 1], [0, 0], [0, 0], [0, 0], [0, 0],
[1, 1], [0, 0], [0, 0], [0, 0], [0, 1], [1, 0],
[0, 0], [1, 0], [0, 0]])
new_features = mdr.transform(test_features)
assert np.array_equal(new_features,
[[0], [0], [1], [1], [1], [1], [0], [1], [1], [1],
[1], [0], [1], [0], [1]])
def test_transform(): """Ensure that the MDR 'transform' method maps a new set of feature instances to the desired labels""" features = np.array([ [2, 0], [0, 0], [0, 1], [0, 0], [0, 0], [0, 0], [0, 1], [0, 0], [0, 0], [0, 1], [0, 0], [0, 0], [0, 0], [1, 1], [1, 1]]) classes = np.array([1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0]) mdr = MDR() mdr.fit(features, classes) test_features = np.array([ [2, 2], [1, 1], [0, 0], [0, 0], [0, 0], [0, 0], [1, 1], [0, 0], [0, 0], [0, 0], [0, 1], [1, 0], [0, 0], [1, 0], [0, 0]]) new_features = mdr.transform(test_features) assert np.array_equal(new_features, [0,0,0,0,0,0,0,0,0,0,1,0,0,0,0])
