def test_add_profilers(self):
compiler1 = col_pro_compilers.BaseCompiler(mock.Mock())
compiler2 = col_pro_compilers.BaseCompiler(mock.Mock())
# test incorrect type
with self.assertRaisesRegex(
TypeError,
"`BaseCompiler` and `int` are "
"not of the same profile compiler type.",
):
compiler1 + 3
compiler3 = col_pro_compilers.ColumnStatsProfileCompiler(mock.Mock())
compiler3._profiles = [mock.Mock()]
with self.assertRaisesRegex(
TypeError,
"`BaseCompiler` and "
"`ColumnStatsProfileCompiler` are "
"not of the same profile compiler type.",
):
compiler1 + compiler3
# test mismatched names
compiler1.name = "compiler1"
compiler2.name = "compiler2"
with self.assertRaisesRegex(
ValueError, "Column profile names are unmatched: "
"compiler1 != compiler2"):
compiler1 + compiler2
# test mismatched profiles due to options
compiler2.name = "compiler1"
compiler1._profiles = dict(test1=mock.Mock())
compiler2._profiles = dict(test2=mock.Mock())
with self.assertRaisesRegex(
ValueError,
"Column profilers were not setup with the "
"same options, hence they do not calculate "
"the same profiles and cannot be added "
"together.",
):
compiler1 + compiler2
# test success
compiler1._profiles = dict(test=1)
compiler2._profiles = dict(test=2)
merged_compiler = compiler1 + compiler2
self.assertEqual(3, merged_compiler._profiles["test"])
self.assertEqual("compiler1", merged_compiler.name)
def test_cannot_instantiate(self):
"""showing we normally can't instantiate an abstract class"""
with self.assertRaises(TypeError) as e:
col_pro_compilers.BaseCompiler()
self.assertEqual(
"Can't instantiate abstract class BaseCompiler with "
"abstract methods profile", str(e.exception))
def test_add_profilers(self):
compiler1 = col_pro_compilers.BaseCompiler(mock.Mock())
compiler2 = col_pro_compilers.BaseCompiler(mock.Mock())
# test incorrect type
with self.assertRaisesRegex(
TypeError, '`BaseCompiler` and `int` are '
'not of the same profile compiler type.'):
compiler1 + 3
compiler3 = col_pro_compilers.ColumnStatsProfileCompiler(mock.Mock())
compiler3._profiles = [mock.Mock()]
with self.assertRaisesRegex(
TypeError, '`BaseCompiler` and '
'`ColumnStatsProfileCompiler` are '
'not of the same profile compiler type.'):
compiler1 + compiler3
# test mismatched names
compiler1.name = 'compiler1'
compiler2.name = 'compiler2'
with self.assertRaisesRegex(
ValueError, 'Column profile names are unmatched: '
'compiler1 != compiler2'):
compiler1 + compiler2
# test mismatched profiles due to options
compiler2.name = 'compiler1'
compiler1._profiles = dict(test1=mock.Mock())
compiler2._profiles = dict(test2=mock.Mock())
with self.assertRaisesRegex(
ValueError, 'Column profilers were not setup with the '
'same options, hence they do not calculate '
'the same profiles and cannot be added '
'together.'):
compiler1 + compiler2
# test success
compiler1._profiles = dict(test=1)
compiler2._profiles = dict(test=2)
merged_compiler = compiler1 + compiler2
self.assertEqual(3, merged_compiler._profiles['test'])
self.assertEqual('compiler1', merged_compiler.name)
def test_no_options_error(self):
with self.assertRaisesRegex(NotImplementedError,
"Must set the expected OptionClass."):
col_pro_compilers.BaseCompiler()
def test_no_profilers_error(self):
with self.assertRaises(NotImplementedError) as e:
col_pro_compilers.BaseCompiler()
self.assertEqual("Must add profilers.", str(e.exception))