单位说法违法，请完三个月的假，不能直接辞退。

医疗期限届满，不能工作，单位可以解除合同。

作为孕妇你只要开具病假单给公司就行，公司如果解除属于违法解除

孕妇不适用医疗期规定，医疗期满以后也不能辞退你，如果辞退你，可以来找律师维权。