def test_basic_feature_matrix(features_test_dir):
basic_path = os.path.join(features_test_dir, 'test_feature_matrix.txt')
with pytest.raises(DelimiterError):
load_feature_matrix_csv('test',basic_path,' ')
fm = load_feature_matrix_csv('test',basic_path,',')
assert(fm.name == 'test')
assert(fm['a','feature1'] == '+')
def test_extra_feature(features_test_dir):
extra_feature_path = os.path.join(features_test_dir, 'test_feature_matrix_extra_feature.txt')
fm = load_feature_matrix_csv('test',extra_feature_path,',')
with pytest.raises(KeyError):
fm.__getitem__(('a','feature3'))
def test_missing_value(features_test_dir):
missing_value_path = os.path.join(features_test_dir, 'test_feature_matrix_missing_value.txt')
fm = load_feature_matrix_csv('test',missing_value_path,',')
assert(fm['d','feature2'] == 'n')
