1 files changed, 21 insertions, 0 deletions
diff --git a/tests/qc/test_header.py b/tests/qc/test_header.py
new file mode 100644
index 0000000..2feac19
--- /dev/null
+++ b/tests/qc/test_header.py
@@ -0,0 +1,21 @@
+import pytest
+from hypothesis import given
+from hypothesis import strategies as st
+
+from quality_control.headers import valid_header
+from quality_control.errors import InvalidHeaderValue
+
+@given(headers=st.lists(st.text(max_size=10)))
+def test_valid_header_errors_with_invalid_headers(headers):
+    "Verify that the check for header validity works"
+    with pytest.raises(InvalidHeaderValue):
+        valid_header(("BXD1", "BXD2", "BXD3"), headers)
+
+@pytest.mark.parametrize(
+    "strains,headers", [
+        (("BXD1", "BXD2", "BXD3"), ("ProbeSet", "BXD3", "BXD1")),
+        (("AStrain", "AnotherStrain", "YetAnotherStrain"),
+         ("Individual", "AStrain", "AnotherStrain", "YetAnotherStrain"))])
+def test_valid_header_strains_passes_with_valid_headers(strains, headers):
+    "Verify that the check for header validity works"
+    assert valid_header(strains, headers)