def test_strfy(self, data, expected): result = du.strfy(data) assert_equal(result, expected)
class TestImport(unittest.TestCase): @parameterized.expand([ ("s", "s"), ("", ""), (1, ""), (dt.datetime.now(), ""), ]) def test_strfy(self, data, expected): result = du.strfy(data) assert_equal(result, expected) def test_concat(self): sheet = { "A39": MockCell("foo"), "B39": MockCell("bar"), "C39": MockCell("baz"), } result = du.concat(["A39", "B39", "C39"])(sheet) assert_equal(result, "foobarbaz") @parameterized.expand([ ("No", "No", " ", ""), ("No", "No", " additional", "additional"), ("No", "Yes", "", "Data Owner approval required"), ("No", "Yes", "additional notes", "Data Owner approval required, additional notes"), ("Yes", "No", " ", "Supervisor approval required"), ("Yes", "Yes", " ", "Supervisor approval required, Data Owner approval required"), ("Yes", "Yes", "additional notes", "Supervisor approval required, Data Owner approval required, additional notes"), ]) def test_access_restrictions(self, sup, owner, addl, expected): sheet = { "B16": MockCell(sup), "D16": MockCell(owner), "B17": MockCell(addl), } result = du.access_restrictions("B16", "D16", "B17")(sheet) assert_equal(result, expected) @parameterized.expand([ (dt.datetime(1989, 03, 11), "1989-03-11"), (dt.date(1989, 03, 11), "1989-03-11"), ("1989-03-11", "1989-03-11"), ("n/a", ""), ]) def test_date(self, cell_val, expected): sheet = {"A1": MockCell(cell_val)} result = du.date("A1")(sheet) assert_equal(result, expected) @parameterized.expand([ ("Yes", "yes"), ("yes", "yes"), ("", ""), (None, ""), ]) def test_lower(self, cell_val, expected): sheet = {"A1": MockCell(cell_val)} result = du.lower("A1")(sheet) assert_equal(result, expected) @parameterized.expand([ ("", ""), ("21", "21"), ("2.1", "2.1"), ("<1mb", "1"), ("foo", ""), ("12-13mb", "1213"), ("12", "12"), ]) def test_transfer_initial_size(self, cell_val, expected): sheet = {"B47": MockCell(cell_val)} result = du.transfer_initial_size(sheet) assert_equal(result, expected) @parameterized.expand([ ("A39", "foo"), (lambda x: x["A39"].value.upper(), "FOO"), ]) def test_get_field(self, cell_or_func, expected): fields = {"test": cell_or_func} sheet = {"A39": MockCell("foo")} result = du.get_field(sheet, "test", fields) assert_equal(result, expected) @parameterized.expand([ ("A39", ({"test": "foo"}, [])), (lambda ws: v.dig_id_validator(du.strfy(ws["A39"].value)), ({}, ["test: Must be in the format DI#####"])), ]) def test_make_rec_from_sheet(self, cell_or_func, expected): fields = {"test": cell_or_func} sheet = {"A39": MockCell("foo")} result = du.make_rec_from_sheet(sheet, fields) assert_equal(result, expected)