def test_sample_windows_from_artifacts_one_result(): artifacts = ['001.p', '002.p', '003.p', '004.p', '005.p'] actual = list(sample_windows_from_artifacts(artifacts, 5)) expected = [ ['001.p', '002.p', '003.p', '004.p', '005.p'], ] assert actual == expected
def test_sample_windows_from_artifacts_no_result(): artifacts = ['001.p', '002.p', '003.p', '004.p'] actual = list(sample_windows_from_artifacts(artifacts, 5)) assert actual == []