def test_raise_error_if_some_columns_are_missing(self):
data = {
"Exp": [1],
"Site": ["s"],
"Hole": [None],
"Core": [None],
"Type": [None],
"Section": [None],
}
df = pd.DataFrame(data)
message = "File does not have the expected columns."
with pytest.raises(ValueError, match=message):
create_sample_name(df)
def test_removes_dash_from_end_of_string_nan_2(self):
data = {
"Exp": [1],
"Site": [np.nan],
"Hole": [np.nan],
"Core": [np.nan],
"Type": ["t"],
"Section": [np.nan],
"A/W": [np.nan],
}
df = pd.DataFrame(data)
data["Sample"] = ["1-t"]
expected = pd.DataFrame(data)
create_sample_name(df)
assert_frame_equal(df, expected)
def test_creates_Sample_string_if_column_are_nan_6(self):
data = {
"Exp": [1],
"Site": [np.nan],
"Hole": [np.nan],
"Core": [np.nan],
"Type": [np.nan],
"Section": [np.nan],
"A/W": [np.nan],
}
df = pd.DataFrame(data, dtype=str)
data["Sample"] = ["1"]
expected = pd.DataFrame(data, dtype=str)
create_sample_name(df)
assert_frame_equal(df, expected)
def test_creates_Sample_string_if_column_are_null_5(self):
data = {
"Exp": [1],
"Site": [None],
"Hole": [None],
"Core": [None],
"Type": [None],
"Section": [None],
"A/W": ["a"],
}
df = pd.DataFrame(data)
data["Sample"] = ["1-a"]
expected = pd.DataFrame(data)
create_sample_name(df)
assert_frame_equal(df, expected)
def test_creates_Sample_string_if_column_are_nan(self):
data = {
"Exp": [1],
"Site": [np.nan],
"Hole": ["h"],
"Core": [2],
"Type": ["t"],
"Section": [3],
"A/W": ["a"],
}
df = pd.DataFrame(data)
data["Sample"] = ["1-h-2t-3-a"]
expected = pd.DataFrame(data)
create_sample_name(df)
assert_frame_equal(df, expected)
def test_returns_Sample_if_all_columns_are_present(self):
data = {
"Exp": [1],
"Site": ["s"],
"Hole": ["h"],
"Core": [2],
"Type": ["t"],
"Section": [3],
"A/W": ["a"],
}
df = pd.DataFrame(data)
data["Sample"] = ["1-sh-2t-3-a"]
expected = pd.DataFrame(data)
create_sample_name(df)
assert_frame_equal(df, expected)
def test_adds_space_extra_sample_data_if_any_aw(self):
data = {
"Exp": [1],
"Site": ["s"],
"Hole": ["h"],
"Core": [2],
"Type": ["t"],
"Section": [3],
"A/W": ["a"],
"Extra Sample ID Data": ["e"],
}
df = pd.DataFrame(data)
data["Sample"] = ["1-sh-2t-3-a e"]
expected = pd.DataFrame(data)
create_sample_name(df)
assert_frame_equal(df, expected)
def test_otherwise_raise_error(self):
df = pd.DataFrame({"foo": [1]})
message = "File does not have the expected columns."
with pytest.raises(ValueError, match=message):
create_sample_name(df)
