def test_uniquecols(): df = pd.DataFrame({'x': [1, 2, 3, 4], 'y': ['a', 'b', 'c', 'd'], 'z': [8] * 4, 'other': ['same']*4}) df2 = pd.DataFrame({'z': [8], 'other': ['same']}) result = uniquecols(df) assert result.equals(df2)