def test_parse_v0_valueerror_bad_type(self): with self.assertRaises(ValueError): migrate_params({ "column": "A", "refine": json.dumps([{ "type": "selec", "column": "A", "content": { "value": "x" } }]), })
def test_parse_v0_valueerror_bad_change_no_content_key(self): with self.assertRaises(ValueError): migrate_params({ "column": "A", "refine": json.dumps([{ "type": "change", "column": "A", "contentx": { "fromVal": "x", "toVal": "y" }, }]), })
def test_parse_v0_valueerror_bad_type(self): with self.assertRaises(ValueError): migrate_params({ 'column': 'A', 'refine': json.dumps([ { 'type': 'selec', 'column': 'A', 'content': { 'value': 'x' } }, ]) })
def test_parse_v0_valueerror_bad_change_no_content_key(self): with self.assertRaises(ValueError): migrate_params({ 'column': 'A', 'refine': json.dumps([ { 'type': 'change', 'column': 'A', 'contentx': { 'fromVal': 'x', 'toVal': 'y' } }, ]) })
def _test_parse_v2(self, column: str, arr: Dict[str, Any], expected: RefineSpec) -> None: result = migrate_params({"column": column, "refine": arr}) self.assertEqual(result["column"], column) refine = result["refine"] self.assertEqual(refine["renames"], expected.renames) self.assertTrue("blackist" not in refine) self.assertEqual(len(refine.keys()), 1)
def _test_parse_v0(self, column: str, arr: List[Dict[str, Any]], expected: RefineSpec) -> None: """ Test that deprecated input is transformed into what the user expects. """ result = migrate_params({"column": column, "refine": json.dumps(arr)}) self.assertEqual(result["column"], column) refine = result["refine"] self.assertEqual(refine["renames"], expected.renames)
def _test_parse_v2(self, column: str, arr: Dict[str, Any], expected: RefineSpec) -> None: result = migrate_params({ 'column': column, 'refine': arr, }) self.assertEqual(result['column'], column) refine = result['refine'] self.assertEqual(refine['renames'], expected.renames) self.assertTrue('blackist' not in refine) self.assertEqual(len(refine.keys()), 1)
def test_parse_v1(self): result = migrate_params({ 'column': 'A', 'refine': '{"renames": {"a": "b"}, "blacklist": []}', }) self.assertEqual(result, { 'column': 'A', 'refine': { 'renames': { 'a': 'b' }, }, })
def test_parse_v1(self): result = migrate_params({ "column": "A", "refine": '{"renames": {"a": "b"}, "blacklist": []}' }) self.assertEqual(result, { "column": "A", "refine": { "renames": { "a": "b" } } })
def test_parse_v1_empty_refine(self): result = migrate_params({'column': 'A', 'refine': ''}) self.assertEqual(result, {'column': 'A', 'refine': {'renames': {}}})
def test_parse_v0_valueerror_not_dict(self): with self.assertRaises(ValueError): migrate_params({'column': 'A', 'refine': json.dumps('A')})
def test_parse_v1_empty_refine(self): result = migrate_params({"column": "A", "refine": ""}) self.assertEqual(result, {"column": "A", "refine": {"renames": {}}})
def test_parse_v0_valueerror_not_dict(self): with self.assertRaises(ValueError): migrate_params({"column": "A", "refine": json.dumps("A")})