核心结论：单位在您工伤休养三个月后要求上班或主动离职的行为，可能涉及违法解除劳动合同或未履行工伤待遇义务，您有权拒绝并主张合法权益。以下从法律角度分情况说明：

1. 关于工伤停工留薪期的认定

根据《工伤保险条例》第三十三条，工伤职工停工留薪期一般不超过12个月，具体期限需由劳动能力鉴定委员会确认。若您仍处于医疗期内（即停工留薪期未满），单位无权要求您返岗或离职，否则构成违法。
建议行动：

• 立即向单位提供医疗机构出具的病假证明，明确仍需休养的时间；
  
• 若单位拒绝，可向当地劳动监察大队投诉或申请劳动仲裁，要求单位继续履行工伤待遇义务。

2. 若停工留薪期已满，单位要求返岗的合法性

若停工留薪期已届满，单位有权要求您返岗，但需满足以下条件：

• 医疗终结：需经劳动能力鉴定委员会确认您已具备劳动能力；
  
• 合理岗位安排：若您因工伤导致部分劳动能力受限，单位应调整至适合的岗位，不得强制要求从事原岗位或高强度工作。
  若单位未满足上述条件：您可拒绝返岗，并主张单位支付伤残津贴（若构成伤残）或继续治疗费用。

3. 单位要求“主动离职”的违法性

无论停工留薪期是否届满，单位均无权以“不返岗即离职”威胁您。若您未主动提出离职，单位单方解除劳动合同需符合《劳动合同法》第三十九条（严重违纪）或第四十条（医疗期满不能从事原工作）等法定情形，否则构成违法解除，您可主张：

• 继续履行合同；或
  
• 双倍经济补偿金（按工作年限计算，每满一年支付2个月工资）。

4. 后续维权步骤

1. 收集证据：保留工伤认定书、病假证明、单位通知返岗或离职的书面记录（如微信、邮件、录音等）；
   
2. 协商沟通：向单位书面说明仍需休养的理由，要求继续享受工伤待遇；
   
3. 申请鉴定：若对停工留薪期有争议，可向劳动能力鉴定委员会申请延长鉴定；
   
4. 投诉或仲裁：若协商无果，向劳动监察部门投诉或直接申请劳动仲裁，主张违法解除赔偿或工伤待遇补足。

总结：单位的要求可能涉嫌违法，您无需被动接受。建议优先通过书面形式与单位沟通，明确拒绝违法要求，并同步准备维权证据。如有进一步疑问，可直接向我咨询具体操作细节。