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')