def test__validate_dataset_structure_raise_error(self): """Test that a ValueError is raised because the bad structure""" # Setup sdv = Mock() sdv.metadata.get_tables.return_value = ['foo', 'bar', 'tar'] sdv.metadata.get_parents.side_effect = [[], [], ['foo', 'bar']] # Run with pytest.raises(ValueError): SDV._validate_dataset_structure(sdv)
def test__validate_dataset_structure_no_error(self): """Test that any error is raised with a supported structure""" # Setup sdv = Mock() sdv.metadata.get_tables.return_value = ['foo', 'bar', 'tar'] sdv.metadata.get_parents.side_effect = [[], ['foo'], ['bar']] # Run SDV._validate_dataset_structure(sdv) # Asserts assert sdv.metadata.get_parents.call_count == 3